| /aosp_15_r20/external/pytorch/c10/core/ |
| H A D | Device.cpp | 13 DeviceType parse_type(const std::string& device_string) { in parse_type() argument
42 [&device_string](const std::pair<const char*, DeviceType>& p) { in parse_type()
43 return p.first && p.first == device_string; in parse_type()
48 if (device_string == get_privateuse1_backend()) { in parse_type()
62 device_string); in parse_type()
68 Device::Device(const std::string& device_string) : Device(Type::CPU) { in Device() argument
69 TORCH_CHECK(!device_string.empty(), "Device string must not be empty"); in Device()
78 pstate != DeviceStringParsingState::ERROR && i < device_string.size(); in Device()
80 const char ch = device_string.at(i); in Device()
126 TORCH_CHECK(!has_error, "Invalid device string: '", device_string, "'"); in Device()
[all …]
|
| H A D | Device.h | 46 /* implicit */ Device(const std::string& device_string);
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | nccl_collective_permute_thunk.cc | 131 std::string device_string = GetDeviceString(params.nccl_params); in RunNcclCollective() local
138 comm, device_string, current_id); in RunNcclCollective()
144 absl::string_view device_string, int64_t current_id) { in RunCollectivePermute() argument
196 device_string, current_id, in RunCollectivePermute()
213 device_string, src_addr.opaque(), element_count, *target_id, in RunCollectivePermute()
224 device_string, dest_addr.opaque(), element_count, *source_id, in RunCollectivePermute()
235 device_string); in RunCollectivePermute()
|
| H A D | nccl_collective_permute_thunk.h | 98 absl::string_view device_string, int64_t current_id);
|
| H A D | jitrt_custom_calls.cc | 2037 std::string device_string = NcclCollectiveThunk::GetDeviceString(params); in operator ()() local
2052 device_string, current_id); in operator ()()
|
| /aosp_15_r20/external/pytorch/c10/test/core/ |
| H A D | Device_test.cpp | 10 std::string device_string; member
37 c10::Device d(ds.device_string); in TEST()
39 << "Device String: " << ds.device_string; in TEST()
41 << "Device String: " << ds.device_string; in TEST()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | constant_op_test.cc | 54 string device_string = "CPU"; in PersistentMemoryTrackingTest() local
57 device_string = "GPU"; in PersistentMemoryTrackingTest()
61 device_string, {}, "/job:worker/replica:0/task:0")); in PersistentMemoryTrackingTest()
|
| /aosp_15_r20/external/libbrillo/brillo/blkdev_utils/ |
| H A D | loop_device.cc | 64 std::string device_string; in GetDeviceNumber() local
69 if (!base::ReadFileToString(device_file, &device_string)) in GetDeviceNumber()
73 device_string, ":", base::TRIM_WHITESPACE, base::SPLIT_WANT_NONEMPTY); in GetDeviceNumber()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | lower_functional_ops.cc | 91 bool IsPropagatableDevice(StringPiece device_string) { in IsPropagatableDevice() argument
93 return DeviceNameUtils::ParseFullName(device_string, &device) && in IsPropagatableDevice()
|
| /aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_torch/ |
| H A D | oss_proxy_executor.cpp | 83 std::string device_string = serialized_arg_val["type"].get<std::string>(); in prefill_stack_with_static_arguments() local
85 device_string += ":" + serialized_arg_val["index"].get<std::string>(); in prefill_stack_with_static_arguments()
88 c10::Device device(device_string); in prefill_stack_with_static_arguments()
|
| /aosp_15_r20/frameworks/av/services/mediametrics/ |
| H A D | AudioPowerUsage.cpp | 95 bool AudioPowerUsage::deviceFromString(const std::string& device_string, int32_t& device) { in deviceFromString() argument
123 auto it = deviceTable.find(device_string); in deviceFromString()
|
| /aosp_15_r20/frameworks/av/services/mediametrics/include/mediametricsservice/ |
| H A D | AudioPowerUsage.h | 88 static bool deviceFromString(const std::string& device_string, int32_t& device);
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/ |
| H A D | tpu_embedding_v2.py | 1066 def check_device(path, device_string): argument
1067 spec = tf_device.DeviceSpec.from_string(device_string)
1075 "enqueue method for an example.".format(path, device_string))
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/training/saving/ |
| H A D | saveable_object_util.py | 53 def set_cpu0(device_string): argument
64 parsed_device = pydev.DeviceSpec.from_string(device_string)
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/ |
| H A D | unpickler.cpp | 809 auto device_string = stack_.back().toTupleRef().elements().at(0); in readGlobal() local
811 stack_.emplace_back(c10::Device(device_string.toStringRef())); in readGlobal()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
| H A D | ops.py | 4796 device_string = device_spec.string_merge(op)
4800 if device_string is not prior_device_string:
4801 op._set_device_from_string(device_string)
4802 prior_device_string = device_string
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| H A D | distributed_tpu_rewrite_pass.cc | 1368 bool IsTpuDevice(StringPiece device_string) { in IsTpuDevice() argument
1370 return DeviceNameUtils::ParseFullName(device_string, &device) && in IsTpuDevice()
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_tensor_creation_ops.py | 1881 device_string = 'cuda:' + str(torch.cuda.current_device())
1882 self.assertEqual(device_string,
|