@Namespace(value="tensorflow") @NoOffset @Properties(inherit=tensorflow.class) public class KernelAndDeviceFunc extends KernelAndDevice
Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator, Pointer.ReferenceCounter
Constructor and Description |
---|
KernelAndDeviceFunc(FunctionLibraryRuntime flr,
ProcessFunctionLibraryRuntime pflr,
DeviceVector input_devices,
IntTensorShapeMap input_tensor_shapes,
DtypeAndPartialTensorShapeIntMap input_resource_dtypes_and_shapes,
Pointer runner,
CollectiveExecutor.Handle collective_executor,
Device host_cpu_device,
BytePointer name,
tensorflow.RendezvousCreator rendezvous_creator) |
KernelAndDeviceFunc(FunctionLibraryRuntime flr,
ProcessFunctionLibraryRuntime pflr,
DeviceVector input_devices,
IntTensorShapeMap input_tensor_shapes,
DtypeAndPartialTensorShapeIntMap input_resource_dtypes_and_shapes,
Pointer runner,
CollectiveExecutor.Handle collective_executor,
Device host_cpu_device,
String name,
tensorflow.RendezvousCreator rendezvous_creator) |
KernelAndDeviceFunc(Pointer p)
Pointer cast constructor.
|
Modifier and Type | Method and Description |
---|---|
Status |
Init(NodeDef ndef,
GraphCollector graph_collector) |
int |
input_type(int i) |
Device |
InputDevice(int i) |
OpKernel |
kernel() |
BytePointer |
name() |
int |
num_inputs() |
int |
num_outputs() |
DataTypeVector |
output_dtypes() |
Device |
OutputDevice(int idx) |
Device |
OutputResourceDevice(int idx) |
Status |
Run(ScopedStepContainer step_container,
TensorValueVector inputs,
TensorVector outputs,
NodeExecStats stats,
StepStats step_stats,
GraphCollector graph_collector,
CancellationManager cancellation_manager) |
Status |
Run(TensorValueVector inputs,
TensorVector outputs,
NodeExecStats stats,
StepStats step_stats,
GraphCollector graph_collector,
CancellationManager cancellation_manager) |
device
address, asBuffer, asByteBuffer, availablePhysicalBytes, calloc, capacity, capacity, close, deallocate, deallocate, deallocateReferences, deallocator, deallocator, equals, fill, formatBytes, free, getDirectBufferAddress, getPointer, getPointer, getPointer, getPointer, hashCode, interruptDeallocatorThread, isNull, isNull, limit, limit, malloc, maxBytes, maxPhysicalBytes, memchr, memcmp, memcpy, memmove, memset, offsetAddress, offsetof, offsetof, parseBytes, physicalBytes, physicalBytesInaccurate, position, position, put, realloc, referenceCount, releaseReference, retainReference, setNull, sizeof, sizeof, toString, totalBytes, totalCount, totalPhysicalBytes, withDeallocator, zero
public KernelAndDeviceFunc(Pointer p)
Pointer(Pointer)
.public KernelAndDeviceFunc(FunctionLibraryRuntime flr, ProcessFunctionLibraryRuntime pflr, @ByVal DeviceVector input_devices, @ByVal IntTensorShapeMap input_tensor_shapes, @ByVal DtypeAndPartialTensorShapeIntMap input_resource_dtypes_and_shapes, @Cast(value="std::function<void(std::function<void()>)>*") Pointer runner, @tensorflow.MoveUniquePtr CollectiveExecutor.Handle collective_executor, Device host_cpu_device, @StdString BytePointer name, @ByVal tensorflow.RendezvousCreator rendezvous_creator)
public KernelAndDeviceFunc(FunctionLibraryRuntime flr, ProcessFunctionLibraryRuntime pflr, @ByVal DeviceVector input_devices, @ByVal IntTensorShapeMap input_tensor_shapes, @ByVal DtypeAndPartialTensorShapeIntMap input_resource_dtypes_and_shapes, @Cast(value="std::function<void(std::function<void()>)>*") Pointer runner, @tensorflow.MoveUniquePtr CollectiveExecutor.Handle collective_executor, Device host_cpu_device, @StdString String name, @ByVal tensorflow.RendezvousCreator rendezvous_creator)
@ByVal public Status Init(@Const @ByRef NodeDef ndef, GraphCollector graph_collector)
Init
in class KernelAndDevice
@ByVal public Status Run(@Const @ByRef TensorValueVector inputs, TensorVector outputs, NodeExecStats stats, StepStats step_stats, GraphCollector graph_collector, CancellationManager cancellation_manager)
Run
in class KernelAndDevice
@ByVal public Status Run(ScopedStepContainer step_container, @Const @ByRef TensorValueVector inputs, TensorVector outputs, NodeExecStats stats, StepStats step_stats, GraphCollector graph_collector, CancellationManager cancellation_manager)
Run
in class KernelAndDevice
@Const public OpKernel kernel()
kernel
in class KernelAndDevice
public Device InputDevice(int i)
InputDevice
in class KernelAndDevice
public Device OutputDevice(int idx)
OutputDevice
in class KernelAndDevice
public Device OutputResourceDevice(int idx)
OutputResourceDevice
in class KernelAndDevice
@Cast(value="tensorflow::DataType") public int input_type(int i)
input_type
in class KernelAndDevice
@Const @ByRef public DataTypeVector output_dtypes()
output_dtypes
in class KernelAndDevice
public int num_inputs()
num_inputs
in class KernelAndDevice
public int num_outputs()
num_outputs
in class KernelAndDevice
@StdString public BytePointer name()
name
in class KernelAndDevice
Copyright © 2022. All rights reserved.