| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A 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
|
| /aosp_15_r20/out/soong/.intermediates/build/make/tools/releasetools/ota_metadata_proto_cc/android_recovery_arm64_armv8-2a_cortex-a55_static/gen/proto/ |
| D | ota_metadata.pb.h | 1286 inline void PartitionState::set_device(int index, const std::string& value) { in set_device() function
1290 inline void PartitionState::set_device(int index, std::string&& value) { in set_device() function
1294 inline void PartitionState::set_device(int index, const char* value) { in set_device() function
1299 inline void PartitionState::set_device(int index, const char* value, size_t size) { in set_device() function
1485 inline void DeviceState::set_device(int index, const std::string& value) { in set_device() function
1489 inline void DeviceState::set_device(int index, std::string&& value) { in set_device() function
1493 inline void DeviceState::set_device(int index, const char* value) { in set_device() function
1498 inline void DeviceState::set_device(int index, const char* value, size_t size) { in set_device() function
|
| /aosp_15_r20/prebuilts/runtime/mainline/platform/sdk/include_gen/external/perfetto/perfetto_protos_perfetto_trace_ftrace_zero_gen_headers/gen/external/perfetto/protos/perfetto/trace/ftrace/ |
| H A D | power.pbzero.h | 53 void set_device(const char* data, size_t size) { in set_device() function
56 void set_device(::protozero::ConstChars chars) { in set_device() function
59 void set_device(std::string value) { in set_device() function
150 void set_device(const char* data, size_t size) { in set_device() function
153 void set_device(::protozero::ConstChars chars) { in set_device() function
156 void set_device(std::string value) { in set_device() function
|
| /aosp_15_r20/out/soong/.intermediates/external/perfetto/perfetto_protos_perfetto_trace_ftrace_zero_gen_headers/gen/external/perfetto/protos/perfetto/trace/ftrace/ |
| D | power.pbzero.h | 53 void set_device(const char* data, size_t size) { in set_device() function
56 void set_device(::protozero::ConstChars chars) { in set_device() function
59 void set_device(std::string value) { in set_device() function
150 void set_device(const char* data, size_t size) { in set_device() function
153 void set_device(::protozero::ConstChars chars) { in set_device() function
156 void set_device(std::string value) { in set_device() function
|
| /aosp_15_r20/external/pytorch/c10/core/impl/ |
| H A 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
|
| /aosp_15_r20/external/pytorch/test/inductor/extension_backends/triton/ |
| H A D | device_interface.py | 55 def set_device(device: int) -> None: member in DeviceInterface.Worker
75 def set_device(device) -> None: member in DeviceInterface
|
| H A D | extension_codegen_backend.py | 42 def set_device(self, device_idx: int) -> str: # noqa: ARG002 unused-argument member in CPUDeviceOpOverrides
|
| /aosp_15_r20/external/pytorch/c10/cuda/ |
| H A D | CUDAGuard.h | 40 void set_device(Device device) { in set_device() function
100 void set_device(Device device) { in set_device() function
|
| H A D | CUDAFunctions.cpp | 132 void set_device(DeviceIndex device) { in set_device() function
|
| /aosp_15_r20/external/pytorch/test/jit/ |
| H A D | test_device_analysis.py | 124 def set_device(x): function
132 def set_device(x, device_name: torch.device): function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/hip/impl/ |
| H A 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
|
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/xpu/ |
| H A D | device_op_overrides.py | 9 def set_device(self, device_idx): member in XPUDeviceOpOverrides
|
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/cuda/ |
| H A D | device_op_overrides.py | 9 def set_device(self, device_idx): member in CUDADeviceOpOverrides
|
| /aosp_15_r20/external/pytorch/torch/utils/ |
| H A D | _device.py | 112 def set_device(device): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | resource_handle.h | 57 void set_device(const std::string& device) { device_ = device; } in set_device() function
|
| /aosp_15_r20/external/ComputeLibrary/src/core/CL/ |
| H A D | CLKernelLibrary.cpp | 73 void CLKernelLibrary::set_device(cl::Device device) in set_device() function in arm_compute::CLKernelLibrary
|
| H A D | CLCompileContext.cpp | 332 void CLCompileContext::set_device(cl::Device device) in set_device() function in arm_compute::CLCompileContext
|
| /aosp_15_r20/external/pytorch/torch/cpu/ |
| H A D | __init__.py | 172 def set_device(device: _device_t) -> None: function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/mps/ |
| H A D | MPSGuardImpl.h | 137 void set_device(Device device) { in set_device() function
|
| /aosp_15_r20/external/pytorch/c10/xpu/ |
| H A D | XPUFunctions.cpp | 190 void set_device(DeviceIndex device) { in set_device() function
|
| /aosp_15_r20/external/pytorch/torch/mtia/ |
| H A D | __init__.py | 183 def set_device(device: _device_t) -> None: function
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/ |
| H A D | README.md | 722 ### set_device subsection
|
| /aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/protobuf/ |
| D | eager_service.pb.h | 4746 void Operation::set_device(ArgT0&& arg0, ArgT... args) { in set_device() function
5431 inline void QueueResponse::set_device(int index, const std::string& value) { in set_device() function
5435 inline void QueueResponse::set_device(int index, std::string&& value) { in set_device() function
5439 inline void QueueResponse::set_device(int index, const char* value) { in set_device() function
5444 inline void QueueResponse::set_device(int index, const char* value, size_t size) { in set_device() function
6986 void SendPackedHandleOp_LocalTensorHandle::set_device(ArgT0&& arg0, ArgT... args) { in set_device() function
|
| /aosp_15_r20/external/pytorch/torch/xpu/ |
| H A D | __init__.py | 192 def set_device(device: _device_t) -> None: function
|
| /aosp_15_r20/external/e2fsprogs/ext2ed/ |
| H A D | general_com.c | 149 void set_device (char *command_line) in set_device() function
|