| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | autocast_mode.h | 63 #define DECLARE_DEPRECATED_AUTOCAST_APIS(name, device_type) … argument
157 c10::DeviceType device_type) { in get_autocast_dispatch_key_from_device_type()
181 inline bool is_autocast_available(c10::DeviceType device_type) { in is_autocast_available()
193 c10::DeviceType device_type) { in get_lower_precision_fp_from_device_type()
270 c10::DeviceType device_type) { in promote_type()
279 c10::DeviceType device_type, in promote_type()
374 c10::DeviceType device_type, in firstarg_is_eligible()
382 c10::DeviceType device_type, in type_from_firstarg()
|
| H A D | autocast_mode.cpp | 9 bool is_autocast_enabled(at::DeviceType device_type) { in is_autocast_enabled()
14 void set_autocast_enabled(at::DeviceType device_type, bool enabled) { in set_autocast_enabled()
100 at::ScalarType get_autocast_dtype(at::DeviceType device_type) { in get_autocast_dtype()
104 void set_autocast_dtype(at::DeviceType device_type, at::ScalarType dtype) { in set_autocast_dtype()
120 Tensor cached_cast(at::ScalarType to_type, const Tensor& arg, DeviceType device_type) { in cached_cast()
|
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/ |
| H A D | DMAConnectivity.cpp | 6 c10::DeviceType device_type, in get_detector_key()
21 c10::DeviceType device_type, in register_detector()
29 c10::DeviceType device_type, in detect()
72 c10::DeviceType device_type, in DMAConnectivity()
80 c10::DeviceType device_type, in register_dma_connectivity_detector()
88 c10::DeviceType device_type, in detect_dma_connectivity()
|
| H A D | SymmetricMemory.cpp | 17 c10::DeviceType device_type, in register_allocator()
23 c10::DeviceType device_type) { in get_allocator()
116 c10::DeviceType device_type, in register_allocator()
123 c10::DeviceType device_type) { in get_allocator()
193 c10::DeviceType device_type, in has_multicast_support()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| H A D | compile_mlir_util.cc | 320 bool CanInlineFunctionsPostLegalization(llvm::StringRef device_type) { in CanInlineFunctionsPostLegalization()
377 mlir::OpPassManager& pm, llvm::StringRef device_type, bool prefer_tf2xla, in CreateConvertMlirToXlaHloPipeline()
460 Status LegalizeToHlo(mlir::ModuleOp module_op, llvm::StringRef device_type, in LegalizeToHlo()
504 llvm::StringRef device_type, in BuildHloFromTfInner()
517 mlir::ModuleOp module_op, llvm::StringRef device_type, in ConvertMLIRToXlaComputation()
568 llvm::StringRef device_type, in BuildHloFromTf()
637 llvm::StringRef device_type, bool use_tuple_args, bool analyse_graph, in CompileMlirToXlaHlo()
667 llvm::StringRef device_type, bool use_tuple_args, bool analyse_graph, in CompileSerializedMlirToXlaHlo()
805 llvm::StringRef device_type, in BuildHloFromModule()
821 llvm::StringRef device_type, bool use_tuple_args, bool analyse_graph, in CompileGraphToXlaHlo()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| H A D | legalize_tf_with_tf2xla.cc | 430 const std::string& device_type) { in CreateDeviceMgr()
442 const std::string& device_type, in RewriteOp()
450 const std::string& device_type, bool is_module_pass) in Tf2XlaRewriter()
722 const std::string& device_type, in Tf2XlaRewritePattern()
759 explicit LegalizeTF(llvm::StringRef device_type, bool prefer_tf2xla) { in LegalizeTF()
781 void PopulateLegalizeTfWithTf2XlaPatterns(llvm::StringRef device_type, in PopulateLegalizeTfWithTf2XlaPatterns()
791 llvm::StringRef device_type, bool prefer_tf2xla) { in createLegalizeTfWithTf2XlaPass()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
| H A D | device_spec.py | 38 def _as_device_str_or_none(device_type): argument
111 device_type=None, argument
279 def device_type(self): member in DeviceSpecV2
370 def _components_to_string(job, replica, task, device_type, device_index): argument
436 def device_type(self, device_type): member in DeviceSpecV1
|
| H A D | config.py | 395 def list_physical_devices(device_type=None): argument
429 def list_logical_devices(device_type=None): argument
465 def get_visible_devices(device_type=None): argument
498 def set_visible_devices(devices, device_type=None): argument
|
| H A D | config_test.py | 601 def testGetMemoryInfoBasic(self, device_type): argument
612 def testGetMemoryUsageSubstring(self, device_type): argument
645 def testPeakMemoryUsage(self, device_type): argument
662 def testResetMemoryStats(self, device_type): argument
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | common_utils.py | 429 op_name, variant_name="", *, decorator=None, device_type=None, dtypes=None argument
441 def xfail(op_name, variant_name="", *, device_type=None, dtypes=None): argument
451 def skip(op_name, variant_name="", *, device_type=None, dtypes=None): argument
494 def decorateForModules(decorator, module_classes, device_type=None, dtypes=None): argument
500 device_type=device_type, argument
538 def tol2(op_name, variant_name, override_dct, *, device_type=None): argument
542 def tol1(op_name, override_dct, *, device_type=None): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | device_factory.cc | 53 bool IsDeviceFactoryEnabled(const string& device_type) { in IsDeviceFactoryEnabled()
67 int32 DeviceFactory::DevicePriority(const string& device_type) { in DevicePriority()
78 bool DeviceFactory::IsPluggableDevice(const string& device_type) { in IsPluggableDevice()
89 void DeviceFactory::Register(const string& device_type, in Register()
113 DeviceFactory* DeviceFactory::GetFactory(const string& device_type) { in GetFactory()
|
| H A D | op_kernel.cc | 258 DeviceType device_type, DeviceBase* device, Allocator* allocator, in OpKernelConstruction()
1243 static string Key(StringPiece op_type, const DeviceType& device_type, in Key()
1371 const DeviceType& device_type, StringPiece node_name, in FindKernelRegistration()
1449 Status FindKernelRegistration(const DeviceType& device_type, in FindKernelRegistration()
1461 bool KernelDefAvailable(const DeviceType& device_type, in KernelDefAvailable()
1472 const DeviceType& device_type, StringPiece node_name, in FindKernelDef()
1513 Status FindKernelDef(const DeviceType& device_type, const NodeDef& node_def, in FindKernelDef()
1533 for (const DeviceType& device_type : prioritized_types) { in SupportedDeviceTypesForNode() local
1564 for (const DeviceType& device_type : prioritized_types) { in SupportedDeviceTypesForNode() local
1585 for (const DeviceType& device_type : prioritized_types) { in SupportedDeviceTypesForNode() local
[all …]
|
| H A D | device_factory.h | 38 static void Register(const std::string& device_type, DeviceFactory* factory, in Register()
158 #define REGISTER_LOCAL_DEVICE_FACTORY(device_type, device_factory, ...) \ argument
162 #define INTERNAL_REGISTER_LOCAL_DEVICE_FACTORY(device_type, device_factory, \ argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/ |
| H A D | ts_eager_fallback.cpp | 21 c10::DeviceType device_type) { in _to_eager()
45 c10::DeviceType device_type) { in to_eager()
74 c10::DeviceType device_type) { in to_eager()
101 c10::DispatchKey dispatch_key(c10::DeviceType device_type) { in dispatch_key()
210 c10::DeviceType device_type) { in ts_eager_fallback()
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_device_type.py | 338 device_type: str = "generic_device_type" variable in DeviceTypeTestBase
546 device_type = "cpu" variable in CPUTestBase
554 device_type = "cuda" variable in CUDATestBase
602 device_type = "lazy" variable in LazyTestBase
621 device_type = "mps" variable in MPSTestBase
643 device_type = "xpu" variable in XPUTestBase
665 device_type = "hpu" variable in HPUTestBase
680 device_type = "privateuse1" variable in PrivateUse1TestBase
1196 def __init__(self, dep, reason, device_type=None): argument
1332 def __init__(self, device_type): argument
[all …]
|
| /aosp_15_r20/external/webrtc/infra/specs/ |
| H A D | client.webrtc.perf.json | 32 "device_type": "shamu", string
68 "device_type": "shamu", string
112 "device_type": "hammerhead", string
148 "device_type": "hammerhead", string
191 "device_type": "walleye", string
226 "device_type": "walleye", string
269 "device_type": "redfin", string
304 "device_type": "redfin", string
348 "device_type": "bullhead", string
384 "device_type": "bullhead", string
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | custom_graph_optimizer_registry.cc | 123 const Creator& optimizer_creator, const std::string& device_type, in RegisterPluginOptimizerOrDie()
138 for (const auto& device_type : device_types) { in PrintPluginConfigsIfConflict() local
156 for (const auto& device_type : device_types) { in PrintPluginConfigsIfConflict() local
182 for (const auto& device_type : device_types) { in GetPluginConfigs() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | ring_reducer_test.cc | 48 const DeviceType& device_type, in GetKernel()
60 std::unique_ptr<OpKernel> GetAdd(DataType dtype, const DeviceType& device_type, in GetAdd()
71 std::unique_ptr<OpKernel> GetDiv(DataType dtype, const DeviceType& device_type, in GetDiv()
85 const TensorShape& shape, const DeviceType& device_type, in Init()
116 void RunTest(DataType dtype, const DeviceType& device_type, int num_workers, in RunTest()
|
| H A D | memory_types.cc | 50 const DeviceType& device_type, const Graph* g, in ProcessMemoryTypes()
95 Status ValidateMemoryTypes(const DeviceType& device_type, const Graph* g) { in ValidateMemoryTypes()
156 Status EnsureMemoryTypes(const DeviceType& device_type, in EnsureMemoryTypes()
217 Status MemoryTypeForOutput(const DeviceType& device_type, const Graph* g, in MemoryTypeForOutput()
|
| /aosp_15_r20/external/pytorch/torch/csrc/utils/ |
| H A D | device_lazy_init.cpp | 15 bool is_device_initialized(at::DeviceType device_type) { in is_device_initialized()
20 void device_lazy_init(at::DeviceType device_type) { in device_lazy_init()
58 void set_requires_device_init(at::DeviceType device_type, bool value) { in set_requires_device_init()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/ |
| H A D | collective_param_resolver_distributed_test.cc | 117 const string& device_type, bool nccl) { in DefineWorkers()
124 void DefineWorker(const string& worker_name, const string& device_type, in DefineWorker()
164 const string& device_type, in DefineCollectiveParams()
181 const string& device_type, in CreateCollectiveParams()
283 const string& device_type, bool nccl, in RestartWorker()
|
| /aosp_15_r20/external/pytorch/c10/core/ |
| H A D | Stream.h | 27 DeviceType device_type; member
104 DeviceType device_type() const noexcept { in device_type() function
155 DeviceType device_type) { in unpack3()
|
| /aosp_15_r20/packages/modules/Bluetooth/system/gd/metrics/linux/ |
| D | metrics.cc | 25 void LogMetricsBondCreateAttempt(RawAddress* addr, uint32_t device_type) {} in LogMetricsBondCreateAttempt()
27 void LogMetricsBondStateChanged(RawAddress* addr, uint32_t device_type, uint32_t status, in LogMetricsBondStateChanged()
30 void LogMetricsDeviceInfoReport(RawAddress* addr, uint32_t device_type, uint32_t class_of_device, in LogMetricsDeviceInfoReport()
|
| /aosp_15_r20/packages/modules/Bluetooth/system/gd/rust/topshim/metrics/ |
| D | metrics_shim.cc | 31 void bond_create_attempt(RawAddress addr, uint32_t device_type) { in bond_create_attempt()
35 void bond_state_changed(RawAddress addr, uint32_t device_type, uint32_t status, uint32_t bond_state, in bond_state_changed()
40 void device_info_report(RawAddress addr, uint32_t device_type, uint32_t class_of_device, in device_info_report()
|
| /aosp_15_r20/external/rust/android-crates-io/crates/virtio-drivers/src/transport/ |
| D | pci.rs | 53 fn device_type(pci_device_id: u16) -> DeviceType { in device_type() function
71 let device_type = device_type(device_function_info.device_id); in virtio_device_type() localVariable
84 device_type: DeviceType, field
113 let device_type = device_type(device_id); in new() localVariable
203 fn device_type(&self) -> DeviceType { in device_type() method
|