org.bytedeco.tensorflow

Class KernelAndDeviceOp

java.lang.Object

org.bytedeco.javacpp.Pointer

org.bytedeco.tensorflow.KernelAndDevice

org.bytedeco.tensorflow.KernelAndDeviceOp

All Implemented Interfaces:

AutoCloseable

@Namespace(value="tensorflow")
 @NoOffset
 @Properties(inherit=tensorflow.class)
public class KernelAndDeviceOp
extends KernelAndDevice

Nested Class Summary

Nested classes/interfaces inherited from class org.bytedeco.javacpp.Pointer

Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator, Pointer.ReferenceCounter

Field Summary

Fields inherited from class org.bytedeco.javacpp.Pointer

address, capacity, limit, position

Constructor Summary

Constructors

Constructor and Description
KernelAndDeviceOp(Pointer p) Pointer cast constructor.
KernelAndDeviceOp(Rendezvous rendez, boolean log_memory, FunctionLibraryRuntime flr, Pointer runner, CollectiveExecutor.Handle collective_executor, Device host_cpu_device)

Method Summary

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)

Methods inherited from class org.bytedeco.tensorflow.KernelAndDevice

device

Methods inherited from class org.bytedeco.javacpp.Pointer

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

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Constructor Detail

KernelAndDeviceOp

public KernelAndDeviceOp(Pointer p)

Pointer cast constructor. Invokes Pointer(Pointer).

KernelAndDeviceOp

public KernelAndDeviceOp(Rendezvous rendez,
                         @Cast(value="bool")
                         boolean log_memory,
                         FunctionLibraryRuntime flr,
                         @Cast(value="std::function<void(std::function<void()>)>*")
                         Pointer runner,
                         @tensorflow.MoveUniquePtr
                         CollectiveExecutor.Handle collective_executor,
                         Device host_cpu_device)

Method Detail

Init

@ByVal
public Status Init(@Const @ByRef
                           NodeDef ndef,
                           GraphCollector graph_collector)

Overrides:

Init in class KernelAndDevice

Run

@ByVal
public Status Run(@Const @ByRef
                          TensorValueVector inputs,
                          TensorVector outputs,
                          NodeExecStats stats,
                          StepStats step_stats,
                          GraphCollector graph_collector,
                          CancellationManager cancellation_manager)

Overrides:

Run in class KernelAndDevice

Run

@ByVal
public Status Run(ScopedStepContainer step_container,
                          @Const @ByRef
                          TensorValueVector inputs,
                          TensorVector outputs,
                          NodeExecStats stats,
                          StepStats step_stats,
                          GraphCollector graph_collector,
                          CancellationManager cancellation_manager)

Overrides:

Run in class KernelAndDevice

kernel

@Const
public OpKernel kernel()

Overrides:

kernel in class KernelAndDevice

InputDevice

public Device InputDevice(int i)

Overrides:

InputDevice in class KernelAndDevice

OutputDevice

public Device OutputDevice(int idx)

Overrides:

OutputDevice in class KernelAndDevice

OutputResourceDevice

public Device OutputResourceDevice(int idx)

Overrides:

OutputResourceDevice in class KernelAndDevice

input_type

@Cast(value="tensorflow::DataType")
public int input_type(int i)

Overrides:

input_type in class KernelAndDevice

output_dtypes

@Const
 @ByRef
public DataTypeVector output_dtypes()

Overrides:

output_dtypes in class KernelAndDevice

num_inputs

public int num_inputs()

Overrides:

num_inputs in class KernelAndDevice

num_outputs

public int num_outputs()

Overrides:

num_outputs in class KernelAndDevice

name

@StdString
public BytePointer name()

Overrides:

name in class KernelAndDevice