Home
last modified time | relevance | path

Searched defs:is_cpu (Results 1 – 15 of 15) sorted by relevance

/external/executorch/runtime/core/portable_type/
Ddevice.h48 bool is_cpu() const noexcept { in is_cpu() function
/external/pytorch/test/distributed/checkpoint/fsdp/
Dtest_fsdp_dsd.py97 def is_cpu(v): function
/external/pytorch/c10/core/
DDevice.h156 bool is_cpu() const noexcept { in is_cpu() function
DTensorImpl.h1093 bool is_cpu() const { in is_cpu() function
/external/pytorch/torch/csrc/jit/passes/
Ddevice_type_analysis.cpp74 bool is_cpu = tensor_type->device() && tensor_type->device()->is_cpu(); in isZerodimCPUTensor() local
Dtensorexpr_fuser.cpp1050 bool is_cpu = device->is_cpu(); in typesAreSupported() local
/external/pytorch/torch/csrc/inductor/aoti_torch/
Doss_proxy_executor.cpp303 OSSProxyExecutor::OSSProxyExecutor(const std::string& json_path, bool is_cpu) { in OSSProxyExecutor()
/external/pytorch/torch/_inductor/codegen/aoti_runtime/
Dinterface.cpp54 bool is_cpu, in AOTInductorModelContainerCreate()
/external/pytorch/test/distributed/checkpoint/
Dtest_state_dict.py455 def is_cpu(v): function
/external/pytorch/aten/src/ATen/core/
DTensorBase.h425 bool is_cpu() const { in is_cpu() function
/external/coreboot/src/device/
Ddevice_util.c913 bool is_cpu(const struct device *cpu) in is_cpu() function
/external/skia/tests/
DSkSLTest.cpp118 static constexpr bool is_cpu(SkSLTestFlags flags) { in is_cpu() function
/external/pytorch/benchmarks/dynamo/
Dcommon.py1384 def is_cpu(self) -> bool: member in OnnxModel
/external/pytorch/torch/_inductor/
Dir.py270 def is_cpu(x): function
/external/mesa3d/src/gallium/drivers/zink/
Dzink_types.h1397 bool is_cpu; member