@Namespace(value="tensorflow") @NoOffset @Properties(inherit=tensorflow.class) public class OpKernelContext extends Pointer
Modifier and Type | Class and Description |
---|---|
static class |
OpKernelContext.Params |
Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator, Pointer.ReferenceCounter
Constructor and Description |
---|
OpKernelContext(OpKernelContext.Params params) |
OpKernelContext(OpKernelContext.Params params,
int num_outputs) |
OpKernelContext(Pointer p)
Pointer cast constructor.
|
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 OpKernelContext(Pointer p)
Pointer(Pointer)
.public OpKernelContext(OpKernelContext.Params params)
public OpKernelContext(OpKernelContext.Params params, int num_outputs)
public Env env()
public int num_inputs()
@ByVal public Status input_dtype(@tensorflow.StringPiece BytePointer name, @Cast(value="tensorflow::DataType*") IntPointer dtype)
@ByVal public Status input_dtype(@tensorflow.StringPiece String name, @Cast(value="tensorflow::DataType*") IntPointer dtype)
public int num_outputs()
@Cast(value="tensorflow::DataType") public int expected_output_dtype(int index)
@ByVal public Status input(@tensorflow.StringPiece BytePointer name, @Cast(value="const tensorflow::Tensor**") PointerPointer tensor)
@ByVal public Status input(@tensorflow.StringPiece BytePointer name, @Const @ByPtrPtr Tensor tensor)
@ByVal public Status input(@tensorflow.StringPiece String name, @Const @ByPtrPtr Tensor tensor)
@ByVal public Status input_list(@tensorflow.StringPiece BytePointer name, OpInputList list)
@ByVal public Status input_list(@tensorflow.StringPiece String name, OpInputList list)
@ByVal public Status input_ref_mutex(@tensorflow.StringPiece BytePointer name, @Cast(value="tensorflow::mutex**") PointerPointer out_mutex)
@ByVal public Status input_ref_mutex(@tensorflow.StringPiece BytePointer name, @Cast(value="tensorflow::mutex**") @ByPtrPtr Pointer out_mutex)
@ByVal public Status input_ref_mutex(@tensorflow.StringPiece String name, @Cast(value="tensorflow::mutex**") @ByPtrPtr Pointer out_mutex)
@ByVal public Status mutable_input(@tensorflow.StringPiece BytePointer name, Tensor tensor, @Cast(value="bool") boolean lock_held)
@ByVal public Status mutable_input(@tensorflow.StringPiece String name, Tensor tensor, @Cast(value="bool") boolean lock_held)
@ByVal public Status mutable_input_list(@tensorflow.StringPiece BytePointer name, OpMutableInputList list)
@ByVal public Status mutable_input_list(@tensorflow.StringPiece String name, OpMutableInputList list)
public void replace_ref_input(int index, @Const @ByRef Tensor tensor, @Cast(value="bool") boolean lock_held)
@ByVal public Status replace_ref_input(@tensorflow.StringPiece BytePointer name, @Const @ByRef Tensor tensor, @Cast(value="bool") boolean lock_held)
@ByVal public Status replace_ref_input(@tensorflow.StringPiece String name, @Const @ByRef Tensor tensor, @Cast(value="bool") boolean lock_held)
public void delete_ref_input(int input_index, @Cast(value="bool") boolean lock_held)
@Cast(value="bool") public boolean ValidateInputsAreSameShape(OpKernel op)
public GraphCollector graph_collector()
public void forward_ref_input_to_ref_output(int input_index, int output_index)
@Cast(value="bool") public boolean forward_input_to_output_with_shape(int input_index, int output_index, @Const @ByRef TensorShape output_shape, @Cast(value="tensorflow::Tensor**") PointerPointer output)
@Cast(value="bool") public boolean forward_input_to_output_with_shape(int input_index, int output_index, @Const @ByRef TensorShape output_shape, @ByPtrPtr Tensor output)
@ByVal public Status forward_input_to_output_with_shape(@tensorflow.StringPiece BytePointer input_name, @tensorflow.StringPiece BytePointer output_name, @Const @ByRef TensorShape output_shape, @Cast(value="tensorflow::Tensor**") PointerPointer output)
@ByVal public Status forward_input_to_output_with_shape(@tensorflow.StringPiece BytePointer input_name, @tensorflow.StringPiece BytePointer output_name, @Const @ByRef TensorShape output_shape, @ByPtrPtr Tensor output)
@ByVal public Status forward_input_to_output_with_shape(@tensorflow.StringPiece String input_name, @tensorflow.StringPiece String output_name, @Const @ByRef TensorShape output_shape, @ByPtrPtr Tensor output)
@UniquePtr public Tensor forward_input(int input_index, int output_index, @Cast(value="tensorflow::DataType") int output_dtype, @Const @ByRef TensorShape output_shape, @Cast(value="tensorflow::MemoryType") int output_memory_type, @Const @ByRef AllocatorAttributes output_attr)
@ByVal public Status forward_input_or_allocate_output(@tensorflow.ArraySlice IntPointer candidate_input_indices, int output_index, @Const @ByRef TensorShape output_shape, @Cast(value="tensorflow::Tensor**") PointerPointer output)
@ByVal public Status forward_input_or_allocate_output(@tensorflow.ArraySlice IntPointer candidate_input_indices, int output_index, @Const @ByRef TensorShape output_shape, @ByPtrPtr Tensor output)
@ByVal public Status forward_input_or_allocate_output(@tensorflow.ArraySlice IntBuffer candidate_input_indices, int output_index, @Const @ByRef TensorShape output_shape, @ByPtrPtr Tensor output)
@ByVal public Status forward_input_or_allocate_output(@tensorflow.ArraySlice int[] candidate_input_indices, int output_index, @Const @ByRef TensorShape output_shape, @ByPtrPtr Tensor output)
@ByVal public Status forward_input_or_allocate_output(@ByVal @Cast(value="tensorflow::gtl::ArraySlice<tensorflow::StringPiece>*") StringPieceVector candidate_input_names, @tensorflow.StringPiece BytePointer output_name, @Const @ByRef TensorShape output_shape, @Cast(value="tensorflow::Tensor**") PointerPointer output)
@ByVal public Status forward_input_or_allocate_output(@ByVal @Cast(value="tensorflow::gtl::ArraySlice<tensorflow::StringPiece>*") StringPieceVector candidate_input_names, @tensorflow.StringPiece BytePointer output_name, @Const @ByRef TensorShape output_shape, @ByPtrPtr Tensor output)
@ByVal public Status forward_input_or_allocate_output(@ByVal @Cast(value="tensorflow::gtl::ArraySlice<tensorflow::StringPiece>*") StringPieceVector candidate_input_names, @tensorflow.StringPiece String output_name, @Const @ByRef TensorShape output_shape, @ByPtrPtr Tensor output)
@ByVal public Status forward_input_or_allocate_temp(@tensorflow.ArraySlice IntPointer candidate_input_indices, @Cast(value="tensorflow::DataType") int type, @Const @ByRef TensorShape shape, @Const @ByRef AllocatorAttributes allocator_attr, Tensor out_temp)
@ByVal public Status forward_input_or_allocate_temp(@tensorflow.ArraySlice IntBuffer candidate_input_indices, @Cast(value="tensorflow::DataType") int type, @Const @ByRef TensorShape shape, @Const @ByRef AllocatorAttributes allocator_attr, Tensor out_temp)
@ByVal public Status forward_input_or_allocate_temp(@tensorflow.ArraySlice int[] candidate_input_indices, @Cast(value="tensorflow::DataType") int type, @Const @ByRef TensorShape shape, @Const @ByRef AllocatorAttributes allocator_attr, Tensor out_temp)
@ByVal public Status forward_input_or_allocate_temp(@tensorflow.ArraySlice IntPointer candidate_input_indices, @Cast(value="tensorflow::DataType") int type, @Const @ByRef TensorShape shape, Tensor out_temp)
@ByVal public Status forward_input_or_allocate_temp(@tensorflow.ArraySlice IntBuffer candidate_input_indices, @Cast(value="tensorflow::DataType") int type, @Const @ByRef TensorShape shape, Tensor out_temp)
@ByVal public Status forward_input_or_allocate_temp(@tensorflow.ArraySlice int[] candidate_input_indices, @Cast(value="tensorflow::DataType") int type, @Const @ByRef TensorShape shape, Tensor out_temp)
@ByVal public Status output_list(@tensorflow.StringPiece BytePointer name, OpOutputList list)
@ByVal public Status output_list(@tensorflow.StringPiece String name, OpOutputList list)
@ByVal public Status allocate_output(int index, @Const @ByRef TensorShape shape, @Cast(value="tensorflow::Tensor**") PointerPointer tensor)
@ByVal public Status allocate_output(int index, @Const @ByRef TensorShape shape, @ByPtrPtr Tensor tensor)
@ByVal public Status allocate_output(@tensorflow.StringPiece BytePointer name, @Const @ByRef TensorShape shape, @Cast(value="tensorflow::Tensor**") PointerPointer tensor)
@ByVal public Status allocate_output(@tensorflow.StringPiece BytePointer name, @Const @ByRef TensorShape shape, @ByPtrPtr Tensor tensor)
@ByVal public Status allocate_output(@tensorflow.StringPiece String name, @Const @ByRef TensorShape shape, @ByPtrPtr Tensor tensor)
@ByVal public Status allocate_output(int index, @Const @ByRef TensorShape shape, @Cast(value="tensorflow::Tensor**") PointerPointer tensor, @ByVal AllocatorAttributes attr)
@ByVal public Status allocate_output(int index, @Const @ByRef TensorShape shape, @ByPtrPtr Tensor tensor, @ByVal AllocatorAttributes attr)
@ByVal public Status allocate_output(@tensorflow.StringPiece BytePointer name, @Const @ByRef TensorShape shape, @Cast(value="tensorflow::Tensor**") PointerPointer tensor, @ByVal AllocatorAttributes attr)
@ByVal public Status allocate_output(@tensorflow.StringPiece BytePointer name, @Const @ByRef TensorShape shape, @ByPtrPtr Tensor tensor, @ByVal AllocatorAttributes attr)
@ByVal public Status allocate_output(@tensorflow.StringPiece String name, @Const @ByRef TensorShape shape, @ByPtrPtr Tensor tensor, @ByVal AllocatorAttributes attr)
@ByVal public Status allocate_temp(@Cast(value="tensorflow::DataType") int type, @Const @ByRef TensorShape shape, Tensor out_temp, @ByVal AllocatorAttributes allocator_attr, @Const @ByRef AllocationAttributes allocation_attr)
@ByVal public Status allocate_temp(@Cast(value="tensorflow::DataType") int type, @Const @ByRef TensorShape shape, Tensor out_temp, @ByVal AllocatorAttributes allocator_attr)
@ByVal public Status allocate_temp(@Cast(value="tensorflow::DataType") int type, @Const @ByRef TensorShape shape, Tensor out_temp)
@ByVal public Status allocate_persistent(@Cast(value="tensorflow::DataType") int type, @Const @ByRef TensorShape shape, PersistentTensor out_persistent, @Cast(value="tensorflow::Tensor**") PointerPointer out_tensor, @ByVal AllocatorAttributes attr)
@ByVal public Status allocate_persistent(@Cast(value="tensorflow::DataType") int type, @Const @ByRef TensorShape shape, PersistentTensor out_persistent, @ByPtrPtr Tensor out_tensor, @ByVal AllocatorAttributes attr)
@ByVal public Status allocate_persistent(@Cast(value="tensorflow::DataType") int type, @Const @ByRef TensorShape shape, PersistentTensor out_persistent, @Cast(value="tensorflow::Tensor**") PointerPointer out_tensor)
@ByVal public Status allocate_persistent(@Cast(value="tensorflow::DataType") int type, @Const @ByRef TensorShape shape, PersistentTensor out_persistent, @ByPtrPtr Tensor out_tensor)
@ByVal public Status set_output(@tensorflow.StringPiece BytePointer name, @Const @ByRef Tensor tensor)
@ByVal public Status set_output(@tensorflow.StringPiece String name, @Const @ByRef Tensor tensor)
@ByVal public Status set_output_ref(@tensorflow.StringPiece BytePointer name, @Cast(value="tensorflow::mutex*") Pointer mu, Tensor tensor_for_ref)
@ByVal public Status set_output_ref(@tensorflow.StringPiece String name, @Cast(value="tensorflow::mutex*") Pointer mu, Tensor tensor_for_ref)
@ByVal public Status mutable_output(@tensorflow.StringPiece BytePointer name, @Cast(value="tensorflow::Tensor**") PointerPointer tensor)
@ByVal public Status mutable_output(@tensorflow.StringPiece BytePointer name, @ByPtrPtr Tensor tensor)
@ByVal public Status mutable_output(@tensorflow.StringPiece String name, @ByPtrPtr Tensor tensor)
public DeviceContext input_device_context(int index)
public DeviceContext op_device_context()
@ByVal public AllocatorAttributes input_alloc_attr(int index)
@ByVal public AllocatorAttributes output_alloc_attr(int index)
@ByVal public WrappedAllocatorVector ConsumeWrappedAllocators()
public Rendezvous rendezvous()
@ByVal public Status create_rendezvous(@Cast(value="const tensorflow::int64") long step_id, @Const DeviceMgr device_mgr, @Cast(value="Rendezvous**") PointerPointer r)
@ByVal public Status create_rendezvous(@Cast(value="const tensorflow::int64") long step_id, @Const DeviceMgr device_mgr, @ByPtrPtr Rendezvous r)
public CollectiveExecutor collective_executor()
public SessionState session_state()
@StdString public BytePointer session_handle()
@Const public SessionMetadata session_metadata()
public TensorStore tensor_store()
public CallFrameInterface call_frame()
public FunctionLibraryRuntime function_library()
public StepStatsCollectorInterface stats_collector()
public ResourceMgr resource_manager()
public TensorSliceReaderCacheWrapper slice_reader_cache()
@Const @ByRef public ThreadPoolDevice eigen_cpu_device()
@ByVal public Status MatchSignature(@ByVal @Cast(value="const tensorflow::DataTypeSlice*") DataTypeVector expected_inputs, @ByVal @Cast(value="const tensorflow::DataTypeSlice*") DataTypeVector expected_outputs)
public CancellationManager cancellation_manager()
@ByVal public FrameAndIter frame_iter()
public DeviceBase device()
public void retrieve_accessed_tensors(@Cast(value="tensorflow::TensorReferenceVector*") TensorValueVector out_vector)
public ScopedStepContainer step_container()
public void CtxFailure(@Cast(value="const char*") BytePointer file, int line, @Const @ByRef Status s)
public void CtxFailureWithWarning(@Cast(value="const char*") BytePointer file, int line, @Const @ByRef Status s)
public void CtxFailureWithWarning(String file, int line, @Const @ByRef Status s)
public Tensor mutable_output(int index)
public void set_output_ref(int index, @Cast(value="tensorflow::mutex*") Pointer mu, Tensor tensor_for_ref)
@ByVal public TensorValue release_output(int index)
public void record_temp_memory_allocation(@Cast(value="tensorflow::int64") long size, @Const @ByRef Tensor t)
public void record_persistent_memory_allocation(@Cast(value="tensorflow::int64") long size, @Cast(value="tensorflow::int64") long alloc_id)
public void record_persistent_memory_allocation(@Cast(value="tensorflow::int64") long size)
@Cast(value="tensorflow::int64") public long persistent_memory_allocated()
@Cast(value="tensorflow::int64*") @StdVector public LongPointer persistent_alloc_ids()
public void clear_recorded_memory()
public void set_record_memory_consumption(@Cast(value="bool") boolean v)
public Allocator get_allocator(@ByVal AllocatorAttributes attr)
Copyright © 2022. All rights reserved.