Searched defs:to_device (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_rma_distributed.cc | 36 const string& key, Device* to_device, in RecvBufCall()
78 const string& key, Device* to_device, DeviceContext* to_device_ctx, in RecvFromPeer()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | base_collective_executor.h | 123 bool peer_is_local, const string& key, Device* to_device, in RecvFromPeer()
|
| D | test_collective_executor_mgr.h | 32 bool peer_is_local, const string& key, Device* to_device, in RecvFromPeer()
|
| D | collective_rma_local.cc | 31 const string& key, Device* to_device, DeviceContext* to_device_ctx, in RecvFromPeer()
|
| D | ring_gatherer_test.cc | 70 bool peer_is_local, const string& key, Device* to_device, in RecvFromPeer()
|
| D | ring_reducer_test.cc | 70 bool peer_is_local, const string& key, Device* to_device, in RecvFromPeer()
|
| D | hierarchical_tree_broadcaster_test.cc | 166 bool peer_is_local, const string& key, Device* to_device, in RecvFromPeer()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_module_group_metadata.cc | 542 auto to_device = GetInstructionDevice(*channel.recv); in DumpCollectedStats() local
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | core_test.py | 1029 def _send(self, tensor, tensor_name, to_device): argument
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_driver.cc | 1421 auto to_device = DeviceFromContext(to); in CanEnablePeerAccess() local
|