| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/ |
| H A 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()
|
| H A D | jax_jit.cc | 1421 -> xla::StatusOr<py::object> { in BuildJaxjitSubmodule()
|
| /aosp_15_r20/external/pytorch/c10/cuda/ |
| H A D | CUDAFunctions.cpp | 252 DeviceIndex ExchangeDevice(DeviceIndex to_device) { in ExchangeDevice()
269 DeviceIndex MaybeExchangeDevice(DeviceIndex to_device) { in MaybeExchangeDevice()
323 DeviceIndex ExchangeDevice(DeviceIndex to_device) { in ExchangeDevice()
333 DeviceIndex MaybeExchangeDevice(DeviceIndex to_device) { in MaybeExchangeDevice()
|
| /aosp_15_r20/external/pytorch/c10/xpu/ |
| H A D | XPUFunctions.cpp | 196 c10::DeviceIndex exchange_device(c10::DeviceIndex to_device) { in exchange_device()
205 c10::DeviceIndex maybe_exchange_device(c10::DeviceIndex to_device) { in maybe_exchange_device()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/ |
| H A 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()
|
| /aosp_15_r20/external/mesa3d/src/gallium/frontends/clover/core/ |
| H A D | queue.cpp | 124 bool to_device = !(flags & CL_MIGRATE_MEM_OBJECT_HOST); in svm_migrate() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | collective_rma_local.cc | 28 const string& key, Device* to_device, DeviceContext* to_device_ctx, in RecvFromPeer()
|
| H A D | collective_test_util.cc | 60 const string& key, Device* to_device, DeviceContext* to_device_ctx, in RecvFromPeer()
|
| /aosp_15_r20/external/mesa3d/src/gallium/drivers/nouveau/nvc0/ |
| H A D | nvc0_context.c | 39 bool to_device, bool mem_undefined) in nvc0_svm_migrate()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_module_group_metadata.cc | 542 auto to_device = GetInstructionDevice(*channel.recv); in DumpCollectedStats() local
|
| /aosp_15_r20/external/pytorch/benchmarks/tensorexpr/ |
| H A D | pt_engine.py | 78 def to_device(self, module, device): member in TorchTensorEngine
|
| /aosp_15_r20/external/pytorch/torch/cuda/ |
| H A D | _memory_viz.py | 448 def to_device(device): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/eager/ |
| H A D | core_test.py | 1074 def _send(self, tensor, tensor_name, to_device): argument
|
| /aosp_15_r20/hardware/libhardware_legacy/audio/ |
| H A D | audio_hw_hal.cpp | 96 audio_devices_t to_device = AUDIO_DEVICE_NONE; in convert_audio_device() local
|
| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | module.cpp | 231 torch::Device to_device, in test_DeviceOrDtypeConversionSkipsUndefinedTensor()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_driver.cc | 1595 auto to_device = DeviceFromContext(to); in CanEnablePeerAccess() local
|
| /aosp_15_r20/external/mesa3d/src/gallium/frontends/rusticl/api/ |
| H A D | memory.rs | 2939 let to_device = !bit_check(flags, CL_MIGRATE_MEM_OBJECT_HOST); in enqueue_svm_migrate_mem() localVariable
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | lowering.py | 684 def to_device(x: TensorBox, device: torch.device, *, copy=False): function
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_jit.py | 4691 def to_device(x): function
|