Searched defs:to_device (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | py_values.cc | 40 PjRtDevice* to_device, in HandlePythonScalar()
80 StatusOr<DevicePutResult> HandlePythonInt(py::handle obj, PjRtDevice* to_device, in HandlePythonInt()
125 StatusOr<DevicePutResult> HandleNumpyScalar(py::handle h, PjRtDevice* to_device, in HandleNumpyScalar()
157 StatusOr<DevicePutResult> HandleNumpyArray(py::handle h, PjRtDevice* to_device, in HandleNumpyArray()
208 PjRtDevice* to_device) { in PyBufferHelper()
223 StatusOr<DevicePutResult> HandlePyBuffer(py::handle obj, PjRtDevice* to_device, in HandlePyBuffer()
230 PjRtDevice* to_device, in HandleDeviceArray()
246 StatusOr<DevicePutResult> DevicePut(py::handle arg, PjRtDevice* to_device, in DevicePut()
|
| D | jax_jit.cc | 1421 -> xla::StatusOr<py::object> { in BuildJaxjitSubmodule()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_rma_distributed.cc | 42 const string& peer_task, const string& key, Device* to_device, in RecvBufCall()
110 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 | collective_test_util.cc | 60 const string& key, Device* to_device, DeviceContext* 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
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | core_test.py | 1074 def _send(self, tensor, tensor_name, to_device): argument
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_driver.cc | 1595 auto to_device = DeviceFromContext(to); in CanEnablePeerAccess() local
|