void |
Device.ConsumeListOfAccessedTensors(DeviceContext context,
TensorValueVector tensors) |
void |
Device.ConsumeListOfAccessedTensors(DeviceContext context,
TensorValueVector tensors) |
void |
DeviceBase.CopyTensorInSameDevice(Tensor input_tensor,
Tensor output_tensor,
DeviceContext device_context,
Pointer done) |
void |
DeviceBase.CopyTensorInSameDevice(Tensor input_tensor,
Tensor output_tensor,
DeviceContext device_context,
Pointer done) |
DeviceBase.GpuDeviceInfo |
DeviceBase.GpuDeviceInfo.default_context(DeviceContext setter) |
DeviceBase.GpuDeviceInfo |
DeviceBase.GpuDeviceInfo.default_context(DeviceContext setter) |
DeviceContextVector.Iterator |
DeviceContextVector.insert(DeviceContextVector.Iterator pos,
DeviceContext value) |
DeviceContextVector.Iterator |
DeviceContextVector.insert(DeviceContextVector.Iterator pos,
DeviceContext value) |
OpKernelContext.Params |
OpKernelContext.Params.op_device_context(DeviceContext setter) |
OpKernelContext.Params |
OpKernelContext.Params.op_device_context(DeviceContext 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) |
DeviceContextVector |
DeviceContextVector.push_back(DeviceContext value) |
DeviceContextVector |
DeviceContextVector.push_back(DeviceContext value) |
DeviceContextVector |
DeviceContextVector.put(DeviceContext... array) |
DeviceContextVector |
DeviceContextVector.put(DeviceContext... array) |
DeviceContextVector |
DeviceContextVector.put(DeviceContext value) |
DeviceContextVector |
DeviceContextVector.put(DeviceContext value) |
DeviceContextInlinedVector |
DeviceContextInlinedVector.put(long i,
DeviceContext value) |
DeviceContextVector |
DeviceContextVector.put(long i,
DeviceContext value) |
DeviceContextVector |
DeviceContextVector.put(long i,
DeviceContext value) |
DeviceContextInlinedVector |
DeviceContextInlinedVector.put(long i,
DeviceContext 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) |
Status |
DeviceBase.ReinitializeGpuDevice(OpKernelContext arg0,
PerOpGpuDevice arg1,
DeviceContext arg2,
Allocator arg3) |
Status |
DeviceBase.ReinitializeGpuDevice(OpKernelContext arg0,
PerOpGpuDevice arg1,
DeviceContext arg2,
Allocator arg3) |
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) |