Searched defs:dev_to_dev_stream_index (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | collective_rma_local.cc | 30 const DeviceLocality& client_locality, int dev_to_dev_stream_index, in RecvFromPeer()
122 int dev_to_dev_stream_index, const StatusCallback& done) { in MemCpyAsync()
|
| H A D | copy_tensor.cc | 129 int dev_to_dev_stream_index, StatusCallback done) { in CopyDeviceToDevice()
207 int dev_to_dev_stream_index, StatusCallback done, in ViaDMA()
|
| H A D | collective_test_util.cc | 62 const DeviceLocality& client_locality, int dev_to_dev_stream_index, in RecvFromPeer()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| H A D | pluggable_device_util.cc | 114 int dev_to_dev_stream_index, StatusCallback done) { in DeviceToDeviceCopy()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/ |
| H A D | collective_rma_distributed.cc | 112 const DeviceLocality& client_locality, int dev_to_dev_stream_index, in RecvFromPeer()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | xla_tpu_device.cc | 129 int dev_to_dev_stream_index, StatusCallback done) { in TpuDeviceToDeviceCopy()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| H A D | gpu_util.cc | 191 int dev_to_dev_stream_index, StatusCallback done) { in DeviceToDeviceCopy()
|