| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_rma_distributed.cc | 38 const string& key, Device* to_device, in RecvBufCall() argument
53 req_.set_dst_device(to_device->name()); in RecvBufCall()
80 const string& key, Device* to_device, DeviceContext* to_device_ctx, in RecvFromPeer() argument
86 peer_device, peer_task, peer_is_local, key, to_device, to_device_ctx, in RecvFromPeer()
101 auto recv_buf_callback = [this, state, peer_task, to_device, to_alloc_attr, in RecvFromPeer()
122 if (to_device->tensorflow_gpu_device_info()) { in RecvFromPeer()
145 to_device, cpu_attr, to_alloc_attr, cpu_tensor, in RecvFromPeer()
172 step_id_, peer_device, peer_task, key, to_device, to_device_ctx, in RecvFromPeer()
|
| D | collective_rma_distributed.h | 41 bool peer_is_local, const string& key, Device* to_device,
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | jax_jit.cc | 570 DevicePutResult HandleBool(py::handle h, xla::PjRtDevice* to_device, in HandleBool() argument
573 h, pyclient.pjrt_client(), to_device), in HandleBool()
577 DevicePutResult HandleInt(py::handle obj, xla::PjRtDevice* to_device, in HandleInt() argument
581 obj, pyclient.pjrt_client(), to_device), in HandleInt()
585 obj, pyclient.pjrt_client(), to_device), in HandleInt()
592 xla::PjRtDevice* to_device, in HandleFloat() argument
597 h, pyclient.pjrt_client(), to_device), in HandleFloat()
601 h, pyclient.pjrt_client(), to_device), in HandleFloat()
608 xla::PjRtDevice* to_device, in HandleComplex() argument
626 nullptr, to_device)), in HandleComplex()
[all …]
|
| D | jax_jit.h | 169 xla::PjRtDevice* to_device,
|
| /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
48 auto consumer_callback = [to_tensor, to_device_ctx, to_device, to_alloc_attr, in RecvFromPeer()
70 to_device, // dst Device in RecvFromPeer()
|
| D | collective_rma_local.h | 41 bool peer_is_local, const string& key, Device* to_device,
|
| D | permuter_test.cc | 76 bool peer_is_local, const string& key, Device* to_device, in RecvFromPeer() argument
84 peer_device, peer_task, peer_is_local, key, to_device, to_device_ctx, in RecvFromPeer()
|
| D | ring_gatherer_test.cc | 68 bool peer_is_local, const string& key, Device* to_device, in RecvFromPeer() argument
77 peer_device, peer_task, peer_is_local, key, to_device, to_device_ctx, in RecvFromPeer()
|
| D | ring_reducer_test.cc | 69 bool peer_is_local, const string& key, Device* to_device, in RecvFromPeer() argument
78 peer_device, peer_task, peer_is_local, key, to_device, to_device_ctx, in RecvFromPeer()
|
| D | hierarchical_tree_broadcaster_test.cc | 166 bool peer_is_local, const string& key, Device* to_device, in RecvFromPeer() argument
174 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 | 542 auto to_device = GetInstructionDevice(*channel.recv); in DumpCollectedStats() local
544 << " to_device=" << *to_device << " send=" << channel.send->name() in DumpCollectedStats()
549 *to_device)] += 1; in DumpCollectedStats()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_driver.cc | 1583 auto to_device = DeviceFromContext(to); in CanEnablePeerAccess() local
1584 if (!to_device.ok()) { in CanEnablePeerAccess()
1586 << to_device.status(); in CanEnablePeerAccess()
1589 return CanEnablePeerAccess(from_device.ValueOrDie(), to_device.ValueOrDie()); in CanEnablePeerAccess()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 266 Device* to_device, DeviceContext* to_device_ctx,
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | core_test.py | 1064 def _send(self, tensor, tensor_name, to_device): argument
1071 'recv_device', to_device,
|