| /aosp_15_r20/external/pytorch/torch/profiler/ |
| H A D | _memory_profiler.py | 998 def _coalesce_timeline(self, device_str): argument
1058 def export_memory_timeline(self, path, device_str) -> None: argument
1069 def export_memory_timeline_raw(self, path, device_str) -> None: argument
1136 self, path, device_str, figsize=(20, 12), title=None argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_runtime/ |
| H A D | model.h | 64 const std::string& device_str, in parse_device_str()
99 const std::string& device_str, in AOTInductorModelBase()
608 const std::string& device_str, in Create()
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/python/ |
| H A D | init.cpp | 30 torch::lazy::BackendDevice GetDeviceOrCurrent(const std::string& device_str) { in GetDeviceOrCurrent()
108 bool wait) { in initLazyBindings()
|
| /aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_runner/ |
| H A D | model_container_runner_cuda.cpp | 9 const std::string& device_str, in AOTIModelContainerRunnerCuda()
|
| H A D | model_container_runner.cpp | 37 const std::string& device_str, in AOTIModelContainerRunner()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | device_util.py | 133 def _set_device_from_string(self, device_str): argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/ |
| H A D | promoted_prim_ops.cpp | 100 std::string device_str = fmt::format("{}:{}", type, index); in device_with_index() local
|
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/aoti_runtime/ |
| H A D | interface.cpp | 66 const char* device_str, in AOTInductorModelContainerCreateWithDevice()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/ |
| H A D | utils.cc | 244 DeviceProperties GetDeviceInfo(const string& device_str) { in GetDeviceInfo()
|
| /aosp_15_r20/external/pytorch/torch/csrc/profiler/standalone/ |
| H A D | execution_trace_observer.cpp | 382 std::string device_str = ""; in convertIValue() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/ |
| H A D | tpu.py | 525 def _set_device_from_string(self, device_str): argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/utils/ |
| H A D | python_arg_parser.h | 827 const std::string& device_str = THPUtils_unpackString(obj); in toDevice() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/core/ |
| H A D | lazy_graph_executor.cpp | 443 for (auto& device_str : DeviceContextArena::Get()->GetActiveDevices()) { in WaitDeviceOps() local
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_fake_tensor.py | 83 def checkType(self, t, device_str, size): argument
|
| /aosp_15_r20/external/selinux/libsepol/src/ |
| H A D | kernel_to_conf.c | 3021 char device_str[20]; /* 2^64-1 <= 16 digits (hex) so < 19 chars */ in write_xen_pcidevice_rules_to_conf() local
|
| H A D | kernel_to_cil.c | 3171 char device_str[20]; /* 2^64-1 <= 16 digits (hex) so (low high) < 19 chars */ in write_xen_pcidevice_rules_to_cil() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | op_kernel.cc | 1483 const std::string device_str = DeviceTypeString(device_type); in FindKernelDef() local
|
| /aosp_15_r20/external/libpcap/ |
| H A D | pcap.c | 2252 char *device_str; local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/ |
| H A D | backend.py | 852 def _set_device_from_string(self, device_str): argument
|
| /aosp_15_r20/external/libcups/scheduler/ |
| H A D | ipp.c | 7186 device_str[256],/* Escaped ppd-device-id string */ in get_ppds() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
| H A D | ops.py | 2372 def _set_device_from_string(self, device_str): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/control_flow/ |
| H A D | control_flow_ops_py_test.py | 727 def _count_matching_switch_nodes_on_device(self, run_metadata, device_str, argument
|