Status |
OpKernelContext.allocate_output(BytePointer name,
TensorShape shape,
PointerPointer tensor,
AllocatorAttributes attr) |
Status |
OpKernelContext.allocate_output(BytePointer name,
TensorShape shape,
PointerPointer tensor,
AllocatorAttributes attr) |
Status |
OpKernelContext.allocate_output(BytePointer name,
TensorShape shape,
Tensor tensor,
AllocatorAttributes attr) |
Status |
OpKernelContext.allocate_output(BytePointer name,
TensorShape shape,
Tensor tensor,
AllocatorAttributes attr) |
Status |
OpKernelContext.allocate_output(int index,
TensorShape shape,
PointerPointer tensor,
AllocatorAttributes attr) |
Status |
OpKernelContext.allocate_output(int index,
TensorShape shape,
PointerPointer tensor,
AllocatorAttributes attr) |
Status |
OpKernelContext.allocate_output(int index,
TensorShape shape,
Tensor tensor,
AllocatorAttributes attr) |
Status |
OpKernelContext.allocate_output(int index,
TensorShape shape,
Tensor tensor,
AllocatorAttributes attr) |
Status |
OpKernelContext.allocate_output(String name,
TensorShape shape,
Tensor tensor,
AllocatorAttributes attr) |
Status |
OpKernelContext.allocate_output(String name,
TensorShape shape,
Tensor tensor,
AllocatorAttributes attr) |
Status |
OpKernelContext.allocate_persistent(int type,
TensorShape shape,
PersistentTensor out_persistent,
PointerPointer out_tensor,
AllocatorAttributes attr) |
Status |
OpKernelContext.allocate_persistent(int type,
TensorShape shape,
PersistentTensor out_persistent,
PointerPointer out_tensor,
AllocatorAttributes attr) |
Status |
OpKernelContext.allocate_persistent(int type,
TensorShape shape,
PersistentTensor out_persistent,
Tensor out_tensor,
AllocatorAttributes attr) |
Status |
OpKernelContext.allocate_persistent(int type,
TensorShape shape,
PersistentTensor out_persistent,
Tensor out_tensor,
AllocatorAttributes attr) |
Status |
OpKernelContext.allocate_temp(int type,
TensorShape shape,
Tensor out_temp,
AllocatorAttributes allocator_attr) |
Status |
OpKernelContext.allocate_temp(int type,
TensorShape shape,
Tensor out_temp,
AllocatorAttributes allocator_attr) |
Status |
OpKernelContext.allocate_temp(int type,
TensorShape shape,
Tensor out_temp,
AllocatorAttributes allocator_attr,
AllocationAttributes allocation_attr) |
Status |
OpKernelContext.allocate_temp(int type,
TensorShape shape,
Tensor out_temp,
AllocatorAttributes allocator_attr,
AllocationAttributes allocation_attr) |
FunctionLibraryRuntime.Options |
FunctionLibraryRuntime.Options.args_alloc_attrs(AllocatorAttributes setter) |
FunctionLibraryRuntime.Options |
FunctionLibraryRuntime.Options.args_alloc_attrs(AllocatorAttributes setter) |
Status |
OpKernelContext.forward_input_or_allocate_temp(int[] candidate_input_indices,
int type,
TensorShape shape,
AllocatorAttributes allocator_attr,
Tensor out_temp) |
Status |
OpKernelContext.forward_input_or_allocate_temp(int[] candidate_input_indices,
int type,
TensorShape shape,
AllocatorAttributes allocator_attr,
Tensor out_temp) |
Status |
OpKernelContext.forward_input_or_allocate_temp(IntBuffer candidate_input_indices,
int type,
TensorShape shape,
AllocatorAttributes allocator_attr,
Tensor out_temp) |
Status |
OpKernelContext.forward_input_or_allocate_temp(IntBuffer candidate_input_indices,
int type,
TensorShape shape,
AllocatorAttributes allocator_attr,
Tensor out_temp) |
Status |
OpKernelContext.forward_input_or_allocate_temp(IntPointer candidate_input_indices,
int type,
TensorShape shape,
AllocatorAttributes allocator_attr,
Tensor out_temp) |
Status |
OpKernelContext.forward_input_or_allocate_temp(IntPointer candidate_input_indices,
int type,
TensorShape shape,
AllocatorAttributes allocator_attr,
Tensor out_temp) |
Tensor |
OpKernelContext.forward_input(int input_index,
int output_index,
int output_dtype,
TensorShape output_shape,
int output_memory_type,
AllocatorAttributes output_attr) |
Tensor |
OpKernelContext.forward_input(int input_index,
int output_index,
int output_dtype,
TensorShape output_shape,
int output_memory_type,
AllocatorAttributes output_attr) |
Allocator |
OpKernelContext.get_allocator(AllocatorAttributes attr) |
Allocator |
OpKernelContext.get_allocator(AllocatorAttributes attr) |
Allocator |
DeviceBase.GetAllocator(AllocatorAttributes arg0) |
Allocator |
DeviceBase.GetAllocator(AllocatorAttributes arg0) |
Allocator |
DeviceBase.GetScopedAllocator(AllocatorAttributes attr,
long step_id) |
Allocator |
DeviceBase.GetScopedAllocator(AllocatorAttributes attr,
long step_id) |
Allocator |
DeviceBase.GetStepAllocator(AllocatorAttributes attr,
ResourceMgr arg1) |
Allocator |
DeviceBase.GetStepAllocator(AllocatorAttributes attr,
ResourceMgr arg1) |
boolean |
AllocatorAttributes.IsEqualOrLessRestrictiveThan(AllocatorAttributes other) |
boolean |
AllocatorAttributes.IsEqualOrLessRestrictiveThan(AllocatorAttributes other) |
Status |
DeviceBase.MakeTensorFromProto(TensorProto tensor_proto,
AllocatorAttributes alloc_attrs,
Tensor tensor) |
Status |
DeviceBase.MakeTensorFromProto(TensorProto tensor_proto,
AllocatorAttributes alloc_attrs,
Tensor tensor) |
void |
AllocatorAttributes.Merge(AllocatorAttributes other) |
void |
AllocatorAttributes.Merge(AllocatorAttributes other) |
OpKernelContext.Params |
OpKernelContext.Params.output_attr_array(AllocatorAttributes setter) |
OpKernelContext.Params |
OpKernelContext.Params.output_attr_array(AllocatorAttributes setter) |
void |
PeerAccessInterface.PostToPeer(BytePointer peer_device,
BytePointer peer_task,
BytePointer key,
Device from_device,
DeviceContext from_device_ctx,
AllocatorAttributes from_alloc_attr,
Tensor from_tensor,
DeviceLocality client_locality,
Pointer done) |
void |
PeerAccessInterface.PostToPeer(BytePointer peer_device,
BytePointer peer_task,
BytePointer key,
Device from_device,
DeviceContext from_device_ctx,
AllocatorAttributes from_alloc_attr,
Tensor from_tensor,
DeviceLocality client_locality,
Pointer done) |
void |
PeerAccessInterface.PostToPeer(String peer_device,
String peer_task,
String key,
Device from_device,
DeviceContext from_device_ctx,
AllocatorAttributes from_alloc_attr,
Tensor from_tensor,
DeviceLocality client_locality,
Pointer done) |
void |
PeerAccessInterface.PostToPeer(String peer_device,
String peer_task,
String key,
Device from_device,
DeviceContext from_device_ctx,
AllocatorAttributes from_alloc_attr,
Tensor from_tensor,
DeviceLocality client_locality,
Pointer done) |
AllocatorAttributesVector |
AllocatorAttributesVector.put(long i,
AllocatorAttributes value) |
AllocatorAttributesVector |
AllocatorAttributesVector.put(long i,
AllocatorAttributes value) |
static void |
ProcessFunctionLibraryRuntime.ReceiveTensorsAsync(BytePointer source_device,
BytePointer target_device,
BytePointer key_prefix,
long src_incarnation,
long num_tensors,
DeviceContext device_context,
AllocatorAttributes alloc_attrs,
Rendezvous rendezvous,
TensorVector received_tensors,
Pointer done) |
static void |
ProcessFunctionLibraryRuntime.ReceiveTensorsAsync(BytePointer source_device,
BytePointer target_device,
BytePointer key_prefix,
long src_incarnation,
long num_tensors,
DeviceContext device_context,
AllocatorAttributes alloc_attrs,
Rendezvous rendezvous,
TensorVector received_tensors,
Pointer done) |
static void |
ProcessFunctionLibraryRuntime.ReceiveTensorsAsync(String source_device,
String target_device,
String key_prefix,
long src_incarnation,
long num_tensors,
DeviceContext device_context,
AllocatorAttributes alloc_attrs,
Rendezvous rendezvous,
TensorVector received_tensors,
Pointer done) |
static void |
ProcessFunctionLibraryRuntime.ReceiveTensorsAsync(String source_device,
String target_device,
String key_prefix,
long src_incarnation,
long num_tensors,
DeviceContext device_context,
AllocatorAttributes alloc_attrs,
Rendezvous rendezvous,
TensorVector received_tensors,
Pointer done) |
void |
PeerAccessInterface.RecvFromPeer(BytePointer peer_device,
BytePointer peer_task,
boolean peer_is_local,
BytePointer key,
Device to_device,
DeviceContext to_device_ctx,
AllocatorAttributes to_alloc_attr,
Tensor to_tensor,
DeviceLocality client_locality,
int dev_to_dev_stream_index,
Pointer done) |
void |
PeerAccessInterface.RecvFromPeer(BytePointer peer_device,
BytePointer peer_task,
boolean peer_is_local,
BytePointer key,
Device to_device,
DeviceContext to_device_ctx,
AllocatorAttributes to_alloc_attr,
Tensor to_tensor,
DeviceLocality client_locality,
int dev_to_dev_stream_index,
Pointer done) |
void |
PeerAccessInterface.RecvFromPeer(String peer_device,
String peer_task,
boolean peer_is_local,
String key,
Device to_device,
DeviceContext to_device_ctx,
AllocatorAttributes to_alloc_attr,
Tensor to_tensor,
DeviceLocality client_locality,
int dev_to_dev_stream_index,
Pointer done) |
void |
PeerAccessInterface.RecvFromPeer(String peer_device,
String peer_task,
boolean peer_is_local,
String key,
Device to_device,
DeviceContext to_device_ctx,
AllocatorAttributes to_alloc_attr,
Tensor to_tensor,
DeviceLocality client_locality,
int dev_to_dev_stream_index,
Pointer done) |
FunctionLibraryRuntime.Options |
FunctionLibraryRuntime.Options.rets_alloc_attrs(AllocatorAttributes setter) |
FunctionLibraryRuntime.Options |
FunctionLibraryRuntime.Options.rets_alloc_attrs(AllocatorAttributes setter) |
static Status |
ProcessFunctionLibraryRuntime.SendTensors(BytePointer source_device,
BytePointer target_device,
BytePointer key_prefix,
long src_incarnation,
TensorVector tensors_to_send,
DeviceContext device_context,
AllocatorAttributes alloc_attrs,
Rendezvous rendezvous) |
static Status |
ProcessFunctionLibraryRuntime.SendTensors(BytePointer source_device,
BytePointer target_device,
BytePointer key_prefix,
long src_incarnation,
TensorVector tensors_to_send,
DeviceContext device_context,
AllocatorAttributes alloc_attrs,
Rendezvous rendezvous) |
static Status |
ProcessFunctionLibraryRuntime.SendTensors(String source_device,
String target_device,
String key_prefix,
long src_incarnation,
TensorVector tensors_to_send,
DeviceContext device_context,
AllocatorAttributes alloc_attrs,
Rendezvous rendezvous) |
static Status |
ProcessFunctionLibraryRuntime.SendTensors(String source_device,
String target_device,
String key_prefix,
long src_incarnation,
TensorVector tensors_to_send,
DeviceContext device_context,
AllocatorAttributes alloc_attrs,
Rendezvous rendezvous) |