| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_rma_local.cc | 30 const DeviceLocality& client_locality, int dev_to_dev_stream_index, in RecvFromPeer() argument
42 dev_to_dev_stream_index, in RecvFromPeer()
58 dev_to_dev_stream_index, [hook, done](const Status& s) { in RecvFromPeer()
87 int dev_to_dev_stream_index, const StatusCallback& done) { in MemCpyAsync() argument
120 dst_attr, src, dst, dev_to_dev_stream_index, done); in MemCpyAsync()
|
| D | copy_tensor.h | 35 Tensor* output, int dev_to_dev_stream_index, StatusCallback done);
48 int dev_to_dev_stream_index, StatusCallback done);
|
| D | copy_tensor.cc | 187 int dev_to_dev_stream_index, StatusCallback done) { in CopyDeviceToDevice() argument
200 dev_to_dev_stream_index](StatusCallback wrapped_done_, in CopyDeviceToDevice()
216 dev_to_dev_stream_index, std::move(wrapped_done_)); in CopyDeviceToDevice()
241 dst_alloc_attr, input, output, dev_to_dev_stream_index, in CopyDeviceToDevice()
254 int dev_to_dev_stream_index, StatusCallback done) { in ViaDMA() argument
284 dev_to_dev_stream_index, std::move(done)); in ViaDMA()
|
| D | collective_rma_local.h | 44 int dev_to_dev_stream_index,
81 int dev_to_dev_stream_index,
|
| D | test_collective_executor_mgr.h | 36 int dev_to_dev_stream_index, in RecvFromPeer() argument
|
| D | ring_gatherer_test.cc | 71 int dev_to_dev_stream_index, in RecvFromPeer() argument
76 to_alloc_attr, to_tensor, client_locality, dev_to_dev_stream_index, in RecvFromPeer()
|
| D | ring_reducer_test.cc | 71 int dev_to_dev_stream_index, in RecvFromPeer() argument
76 to_alloc_attr, to_tensor, client_locality, dev_to_dev_stream_index, in RecvFromPeer()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_rma_distributed.cc | 79 const DeviceLocality& client_locality, int dev_to_dev_stream_index, in RecvFromPeer() argument
84 to_alloc_attr, to_tensor, client_locality, dev_to_dev_stream_index, in RecvFromPeer()
99 to_device_ctx, to_tensor, dev_to_dev_stream_index, in RecvFromPeer()
138 to_tensor, dev_to_dev_stream_index, in RecvFromPeer()
|
| D | collective_rma_distributed.h | 40 int dev_to_dev_stream_index,
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_collective_executor_mgr.cc | 83 int dev_to_dev_stream_index, in RecvFromPeer() argument
88 to_alloc_attr, to_tensor, client_locality, dev_to_dev_stream_index, in RecvFromPeer()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util.h | 97 int dev_to_dev_stream_index, StatusCallback done);
|
| D | gpu_util.cc | 192 int dev_to_dev_stream_index, StatusCallback done) { in DeviceToDeviceCopy() argument
203 ->device_to_device_stream(dev_to_dev_stream_index); in DeviceToDeviceCopy()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 241 int dev_to_dev_stream_index,
|