Searched defs:dev_to_dev_stream_index (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_rma_local.cc | 30 const DeviceLocality& client_locality, int dev_to_dev_stream_index, in RecvFromPeer()
87 int dev_to_dev_stream_index, const StatusCallback& done) { in MemCpyAsync()
|
| D | test_collective_executor_mgr.h | 36 int dev_to_dev_stream_index, in RecvFromPeer()
|
| D | copy_tensor.cc | 187 int dev_to_dev_stream_index, StatusCallback done) { in CopyDeviceToDevice()
254 int dev_to_dev_stream_index, StatusCallback done) { in ViaDMA()
|
| D | ring_gatherer_test.cc | 71 int dev_to_dev_stream_index, in RecvFromPeer()
|
| D | ring_reducer_test.cc | 71 int 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()
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_collective_executor_mgr.cc | 83 int dev_to_dev_stream_index, in RecvFromPeer()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util.cc | 192 int dev_to_dev_stream_index, StatusCallback done) { in DeviceToDeviceCopy()
|