Searched defs:to_device (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | jax_jit.cc | 570 DevicePutResult HandleBool(py::handle h, xla::PjRtDevice* to_device, in HandleBool()
577 DevicePutResult HandleInt(py::handle obj, xla::PjRtDevice* to_device, in HandleInt()
592 xla::PjRtDevice* to_device, in HandleFloat()
608 xla::PjRtDevice* to_device, in HandleComplex()
641 xla::PjRtDevice* to_device, in HandleDeviceArray()
663 DevicePutResult HandleBufferFromPyval(py::handle h, xla::PjRtDevice* to_device, in HandleBufferFromPyval()
674 DevicePutResult HandleNpBool(py::handle h, xla::PjRtDevice* to_device, in HandleNpBool()
687 DevicePutResult HandleUint64(py::handle h, xla::PjRtDevice* to_device, in HandleUint64()
711 xla::PjRtDevice* to_device, in HandleNdarray()
741 xla::PjRtDevice* to_device, in DevicePut()
[all …]
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_rma_distributed.cc | 38 const string& key, Device* to_device, in RecvBufCall()
80 const string& key, Device* to_device, DeviceContext* 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()
|
| D | permuter_test.cc | 76 bool peer_is_local, const string& key, Device* to_device, in RecvFromPeer()
|
| D | ring_gatherer_test.cc | 68 bool peer_is_local, const string& key, Device* to_device, in RecvFromPeer()
|
| D | ring_reducer_test.cc | 69 bool peer_is_local, const string& key, Device* to_device, in RecvFromPeer()
|
| D | hierarchical_tree_broadcaster_test.cc | 166 bool peer_is_local, const string& key, Device* to_device, in RecvFromPeer()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_module_group_metadata.cc | 542 auto to_device = GetInstructionDevice(*channel.recv); in DumpCollectedStats() local
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_driver.cc | 1583 auto to_device = DeviceFromContext(to); in CanEnablePeerAccess() local
|