| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_rma_distributed.cc | 36 const string& key, Device* to_device, in RecvBufCall() argument
50 req_.set_dst_device(to_device->name()); in RecvBufCall()
77 const string& key, Device* to_device, DeviceContext* to_device_ctx, in RecvFromPeer() argument
83 peer_device, peer_task, peer_is_local, key, to_device, to_device_ctx, in RecvFromPeer()
98 auto recv_buf_callback = [this, state, peer_task, to_device, to_alloc_attr, in RecvFromPeer()
118 if (to_device->tensorflow_gpu_device_info()) { in RecvFromPeer()
137 to_device, cpu_attr, to_alloc_attr, cpu_tensor, in RecvFromPeer()
163 to_device, to_device_ctx, to_alloc_attr, in RecvFromPeer()
170 step_id_, peer_device, peer_task, key, to_device, to_device_ctx, in RecvFromPeer()
|
| D | collective_rma_distributed.h | 36 bool peer_is_local, const string& key, Device* to_device,
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_collective_executor_mgr.cc | 37 const string& key, Device* to_device, in RecvBufCall() argument
51 req_.set_dst_device(to_device->name()); in RecvBufCall()
79 bool peer_is_local, const string& key, Device* to_device, in RecvFromPeer() argument
87 peer_device, peer_task, peer_is_local, key, to_device, to_device_ctx, in RecvFromPeer()
102 auto recv_buf_callback = [this, state, peer_task, to_device, to_alloc_attr, in RecvFromPeer()
106 to_tensor, state->call->resp_.transport_options(), to_device, in RecvFromPeer()
119 to_device, to_device_ctx, to_alloc_attr, in RecvFromPeer()
126 step_id_, peer_device, peer_task, key, to_device, to_device_ctx, in RecvFromPeer()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_rma_local.cc | 28 const string& key, Device* to_device, DeviceContext* to_device_ctx, in RecvFromPeer() argument
41 key, [to_tensor, to_device_ctx, to_device, to_alloc_attr, in RecvFromPeer()
53 to_device, // dst Device in RecvFromPeer()
|
| D | base_collective_executor.h | 122 bool peer_is_local, const string& key, Device* to_device, in RecvFromPeer() argument
128 peer_device, peer_task, peer_is_local, key, to_device, to_device_ctx, in RecvFromPeer()
|
| D | collective_rma_local.h | 40 bool peer_is_local, const string& key, Device* to_device,
|
| D | test_collective_executor_mgr.h | 32 bool peer_is_local, const string& key, Device* to_device, in RecvFromPeer() argument
|
| D | ring_gatherer_test.cc | 67 bool peer_is_local, const string& key, Device* to_device, in RecvFromPeer() argument
75 peer_device, peer_task, peer_is_local, key, to_device, to_device_ctx, in RecvFromPeer()
|
| D | ring_reducer_test.cc | 67 bool peer_is_local, const string& key, Device* to_device, in RecvFromPeer() argument
75 peer_device, peer_task, peer_is_local, key, to_device, to_device_ctx, in RecvFromPeer()
|
| D | hierarchical_tree_broadcaster_test.cc | 163 bool peer_is_local, const string& key, Device* to_device, in RecvFromPeer() argument
170 peer_device, peer_task, peer_is_local, key, to_device, to_device_ctx, in RecvFromPeer()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_module_group_metadata.cc | 524 auto to_device = GetInstructionDevice(*channel.recv); in DumpCollectedStats() local
526 << " to_device=" << *to_device << " send=" << channel.send->name() in DumpCollectedStats()
531 *to_device)] += 1; in DumpCollectedStats()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_driver.cc | 1577 auto to_device = DeviceFromContext(to); in CanEnablePeerAccess() local
1578 if (!to_device.ok()) { in CanEnablePeerAccess()
1580 << to_device.status(); in CanEnablePeerAccess()
1584 &can_access_peer, from_device.ValueOrDie(), to_device.ValueOrDie()); in CanEnablePeerAccess()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 237 Device* to_device, DeviceContext* to_device_ctx,
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | core_test.py | 707 def _send(self, tensor, tensor_name, to_device): argument
714 'recv_device', to_device,
|