Searched defs:to_device (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_rma_distributed.cc | 36 const string& key, Device* to_device, in RecvBufCall()
77 const string& key, Device* to_device, DeviceContext* to_device_ctx, in RecvFromPeer()
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_collective_executor_mgr.cc | 37 const string& key, Device* to_device, in RecvBufCall()
79 bool peer_is_local, const string& key, Device* to_device, in RecvFromPeer()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | base_collective_executor.h | 122 bool peer_is_local, const string& key, Device* to_device, in RecvFromPeer()
|
| D | collective_rma_local.cc | 28 const string& key, Device* to_device, DeviceContext* to_device_ctx, in RecvFromPeer()
|
| D | test_collective_executor_mgr.h | 32 bool peer_is_local, const string& key, Device* to_device, in RecvFromPeer()
|
| D | ring_gatherer_test.cc | 67 bool peer_is_local, const string& key, Device* to_device, in RecvFromPeer()
|
| D | ring_reducer_test.cc | 67 bool peer_is_local, const string& key, Device* to_device, in RecvFromPeer()
|
| D | hierarchical_tree_broadcaster_test.cc | 163 bool peer_is_local, const string& key, Device* to_device, in RecvFromPeer()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_module_group_metadata.cc | 524 auto to_device = GetInstructionDevice(*channel.recv); in DumpCollectedStats() local
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | core_test.py | 707 def _send(self, tensor, tensor_name, to_device): argument
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_driver.cc | 1577 auto to_device = DeviceFromContext(to); in CanEnablePeerAccess() local
|