| /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() argument
49 dev_to_dev_stream_index, in RecvFromPeer()
75 dev_to_dev_stream_index, in RecvFromPeer()
122 int dev_to_dev_stream_index, const StatusCallback& done) { in MemCpyAsync() argument
155 dst_attr, src, dst, dev_to_dev_stream_index, done); in MemCpyAsync()
|
| H A D | copy_tensor.cc | 129 int dev_to_dev_stream_index, StatusCallback done) { in CopyDeviceToDevice() argument
141 out_allocator, status_cb, dev_to_dev_stream_index, in CopyDeviceToDevice()
150 dev_to_dev_stream_index, wrapped_done); in CopyDeviceToDevice()
166 dev_to_dev_stream_index, wrapped_done); in CopyDeviceToDevice()
194 dst_alloc_attr, input, output, dev_to_dev_stream_index, in CopyDeviceToDevice()
207 int dev_to_dev_stream_index, StatusCallback done, in ViaDMA() argument
244 dev_to_dev_stream_index, std::move(done)); in ViaDMA()
|
| H A D | copy_tensor.h | 35 Tensor* output, int dev_to_dev_stream_index, StatusCallback done);
48 int dev_to_dev_stream_index, StatusCallback done,
|
| H A D | collective_rma_local.h | 45 int dev_to_dev_stream_index,
71 int dev_to_dev_stream_index,
|
| H A D | collective_test_util.cc | 62 const DeviceLocality& client_locality, int dev_to_dev_stream_index, in RecvFromPeer() argument
67 to_alloc_attr, to_tensor, client_locality, dev_to_dev_stream_index, in RecvFromPeer()
|
| H A D | collective_test_util.h | 54 int dev_to_dev_stream_index,
|
| /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() argument
117 to_alloc_attr, to_tensor, client_locality, dev_to_dev_stream_index, in RecvFromPeer()
172 dev_to_dev_stream_index, dst_tensor, done](const Status& s) { in RecvFromPeer()
200 to_tensor, dev_to_dev_stream_index, in RecvFromPeer()
|
| H A D | collective_rma_distributed.h | 45 int dev_to_dev_stream_index,
|
| /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() argument
126 ->device_to_device_stream(dev_to_dev_stream_index); in DeviceToDeviceCopy()
|
| H A D | pluggable_device_util.h | 58 int dev_to_dev_stream_index, StatusCallback done);
|
| /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() argument
202 ->device_to_device_stream(dev_to_dev_stream_index); in DeviceToDeviceCopy()
|
| H A D | gpu_util.h | 97 int dev_to_dev_stream_index, StatusCallback done);
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | collective.h | 285 int dev_to_dev_stream_index,
|
| /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() argument
|