| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_rma_local.cc | 41 Device* from_device; in RecvFromPeer() local
42 Status status = dev_mgr_->LookupDevice(peer_device, &from_device); in RecvFromPeer()
92 buf_rendezvous_.ConsumeBuf(key, from_device->name(), in RecvFromPeer()
93 from_device->attributes().incarnation(), in RecvFromPeer()
99 Device* from_device, DeviceContext* from_device_ctx, in PostToPeer() argument
105 buf_rendezvous_.ProvideBuf(key, from_device, from_device_ctx, from_tensor, in PostToPeer()
|
| D | collective_rma_local.h | 50 const string& key, Device* from_device,
|
| D | permuter_test.cc | 90 const string& key, Device* from_device, in PostToPeer() argument
99 peer_device, peer_task, key, from_device, from_device_ctx, in PostToPeer()
|
| D | ring_gatherer_test.cc | 83 const string& key, Device* from_device, in PostToPeer() argument
92 peer_device, peer_task, key, from_device, from_device_ctx, in PostToPeer()
|
| D | ring_reducer_test.cc | 84 const string& key, Device* from_device, in PostToPeer() argument
93 peer_device, peer_task, key, from_device, from_device_ctx, in PostToPeer()
|
| D | hierarchical_tree_broadcaster_test.cc | 180 const string& key, Device* from_device, in PostToPeer() argument
189 peer_device, peer_task, key, from_device, from_device_ctx, in PostToPeer()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_module_group_metadata.cc | 541 auto from_device = GetInstructionDevice(*channel.send); in DumpCollectedStats() local
543 LOG(INFO) << "Channel " << channel.id << ": from_device=" << *from_device in DumpCollectedStats()
548 communication_histogram[std::pair<int64, int64>(*from_device, in DumpCollectedStats()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_driver.cc | 1577 auto from_device = DeviceFromContext(from); in CanEnablePeerAccess() local
1578 if (!from_device.ok()) { in CanEnablePeerAccess()
1580 << from_device.status(); in CanEnablePeerAccess()
1589 return CanEnablePeerAccess(from_device.ValueOrDie(), to_device.ValueOrDie()); in CanEnablePeerAccess()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 275 const string& key, Device* from_device,
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | core_test.py | 1074 def _recv(self, dtype, tensor_name, from_device): argument
1082 'send_device', from_device,
|