| /external/pytorch/torch/_dynamo/ |
| D | device_interface.py | 55 def set_device(device: int): member in DeviceInterface.Worker
71 def set_device(device: _device_t): member in DeviceInterface
164 def set_device(device: int): member in CudaInterface.Worker
194 set_device = staticmethod(torch.cuda.set_device) variable in CudaInterface
235 def set_device(device: int): member in XpuInterface.Worker
265 set_device = staticmethod(torch.xpu.set_device) variable in XpuInterface
|
| /external/pytorch/c10/core/impl/ |
| D | InlineDeviceGuard.h | 120 void set_device(at::Device device) { in set_device() function
347 void set_device(at::Device device) { in set_device() function
|
| /external/pytorch/test/inductor/extension_backends/triton/ |
| D | device_interface.py | 55 def set_device(device: int) -> None: member in DeviceInterface.Worker
75 def set_device(device) -> None: member in DeviceInterface
|
| D | extension_codegen_backend.py | 42 def set_device(self, device_idx: int) -> str: # noqa: ARG002 unused-argument member in CPUDeviceOpOverrides
|
| /external/pytorch/torch/_inductor/codegen/cuda/ |
| D | device_op_overrides.py | 9 def set_device(self, device_idx): member in CUDADeviceOpOverrides
|
| /external/pytorch/torch/_inductor/codegen/xpu/ |
| D | device_op_overrides.py | 9 def set_device(self, device_idx): member in XPUDeviceOpOverrides
|
| /external/pytorch/torch/utils/ |
| D | _device.py | 112 def set_device(device): function
|
| /external/pytorch/c10/cuda/ |
| D | CUDAGuard.h | 40 void set_device(Device device) { in set_device() function
100 void set_device(Device device) { in set_device() function
|
| D | CUDAFunctions.cpp | 132 void set_device(DeviceIndex device) { in set_device() function
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | resource_handle.h | 57 void set_device(const std::string& device) { device_ = device; } in set_device() function
|
| /external/pytorch/test/jit/ |
| D | test_device_analysis.py | 124 def set_device(x): function
132 def set_device(x, device_name: torch.device): function
|
| /external/ComputeLibrary/src/core/CL/ |
| D | CLKernelLibrary.cpp | 73 void CLKernelLibrary::set_device(cl::Device device) in set_device() function in arm_compute::CLKernelLibrary
|
| D | CLCompileContext.cpp | 332 void CLCompileContext::set_device(cl::Device device) in set_device() function in arm_compute::CLCompileContext
|
| /external/pytorch/torch/cpu/ |
| D | __init__.py | 172 def set_device(device: _device_t) -> None: function
|
| /external/pytorch/aten/src/ATen/mps/ |
| D | MPSGuardImpl.h | 137 void set_device(Device device) { in set_device() function
|
| /external/pytorch/aten/src/ATen/hip/impl/ |
| D | HIPGuardImplMasqueradingAsCUDA.h | 261 void set_device(Device device) { guard_.set_device(device); } in set_device() function
281 void set_device(Device device) { guard_.set_device(device); } in set_device() function
|
| /external/pytorch/c10/xpu/ |
| D | XPUFunctions.cpp | 190 void set_device(DeviceIndex device) { in set_device() function
|
| /external/pytorch/torch/mtia/ |
| D | __init__.py | 183 def set_device(device: _device_t) -> None: function
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | README.md | 722 ### set_device subsection
|
| /external/pytorch/torch/xpu/ |
| D | __init__.py | 192 def set_device(device: _device_t) -> None: function
|
| /external/e2fsprogs/ext2ed/ |
| D | general_com.c | 149 void set_device (char *command_line) in set_device() function
|
| /external/pytorch/c10/core/ |
| D | TensorOptions.h | 469 void set_device(std::optional<Device> device) & noexcept { in set_device() function
|
| /external/pytorch/torch/cuda/ |
| D | __init__.py | 466 def set_device(device: _device_t) -> None: function
|
| /external/pytorch/torch/csrc/autograd/ |
| D | engine.cpp | 835 void set_device(int device) { in set_device() function
|
| /external/pytorch/torch/_inductor/codegen/ |
| D | common.py | 103 def set_device(self, device_idx): member in DeviceOpOverrides
|