| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_rma_local.cc | 73 Device* from_device, DeviceContext* from_device_ctx, in PostToPeer() argument
78 buf_rendezvous_.ProvideBuf(key, from_device, from_device_ctx, from_tensor, in PostToPeer()
|
| D | base_collective_executor.h | 133 const string& key, Device* from_device, in PostToPeer() argument
139 remote_access_->PostToPeer(peer_device, peer_task, key, from_device, in PostToPeer()
|
| D | collective_rma_local.h | 48 const string& key, Device* from_device,
|
| D | test_collective_executor_mgr.h | 42 const string& key, Device* from_device, in PostToPeer() argument
|
| D | ring_gatherer_test.cc | 81 const string& key, Device* from_device, in PostToPeer() argument
89 peer_device, peer_task, key, from_device, from_device_ctx, in PostToPeer()
|
| D | ring_reducer_test.cc | 81 const string& key, Device* from_device, in PostToPeer() argument
89 peer_device, peer_task, key, from_device, from_device_ctx, in PostToPeer()
|
| D | hierarchical_tree_broadcaster_test.cc | 175 const string& key, Device* from_device, in PostToPeer() argument
183 peer_device, peer_task, key, from_device, from_device_ctx, in PostToPeer()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_module_group_metadata.cc | 523 auto from_device = GetInstructionDevice(*channel.send); in DumpCollectedStats() local
525 LOG(INFO) << "Channel " << channel.id << ": from_device=" << *from_device in DumpCollectedStats()
530 communication_histogram[std::pair<int64, int64>(*from_device, in DumpCollectedStats()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_driver.cc | 1571 auto from_device = DeviceFromContext(from); in CanEnablePeerAccess() local
1572 if (!from_device.ok()) { in CanEnablePeerAccess()
1574 << from_device.status(); in CanEnablePeerAccess()
1584 &can_access_peer, from_device.ValueOrDie(), to_device.ValueOrDie()); in CanEnablePeerAccess()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 245 const string& key, Device* from_device,
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | core_test.py | 717 def _recv(self, dtype, tensor_name, from_device): argument
725 'send_device', from_device,
|