Searched refs:get_privateuse1_backend (Results 1 – 15 of 15) sorted by relevance
| /external/pytorch/c10/test/core/ |
| D | Device_test.cpp | 54 ASSERT_EQ(c10::get_privateuse1_backend(true), "my_privateuse1_backend"); in TEST()
55 ASSERT_EQ(c10::get_privateuse1_backend(false), "MY_PRIVATEUSE1_BACKEND"); in TEST()
|
| /external/pytorch/c10/core/ |
| D | DeviceType.cpp | 54 return get_privateuse1_backend(/*lower_case=*/lower_case); in DeviceTypeName()
126 std::string get_privateuse1_backend(bool lower_case) { in get_privateuse1_backend() function
|
| D | DeviceType.h | 105 C10_API std::string get_privateuse1_backend(bool lower_case = true);
|
| D | Device.cpp | 48 if (device_string == get_privateuse1_backend()) { in parse_type()
|
| /external/pytorch/aten/src/ATen/core/ |
| D | Tensor.cpp | 89 dispatchkey_str = c10::get_privateuse1_backend(); in toString()
91 dispatchkey_str = "Autocast" + c10::get_privateuse1_backend(); in toString()
|
| D | NamedTensor.cpp | 91 …"NYI: named tensors only support CPU, CUDA, XPU or ", c10::get_privateuse1_backend(), " tensors."); in internal_set_names_inplace()
|
| /external/pytorch/torch/csrc/profiler/ |
| D | kineto_shim.cpp | 363 if (c10::get_privateuse1_backend() != "privateuseone") { in deviceTypeFromActivity()
377 if (c10::get_privateuse1_backend() != "privateuseone") { in deviceTypeFromActivity()
|
| /external/pytorch/torch/csrc/utils/ |
| D | tensor_types.cpp | 22 static std::string backend_name = "torch." + get_privateuse1_backend(); in parse_privateuseone_backend()
|
| /external/pytorch/aten/src/ATen/core/dispatch/ |
| D | OperatorEntry.cpp | 519 auto privateuse1_backend = get_privateuse1_backend(); in registerKernel()
525 auto backend_name = c10::get_privateuse1_backend(); in registerKernel()
|
| /external/pytorch/aten/src/ATen/ |
| D | NestedTensorImpl.cpp | 184 …"NestedTensorImpl storage must be either CUDA, CPU, XPU or ", get_privateuse1_backend(), " but got… in NestedTensorImpl()
|
| /external/pytorch/torch/csrc/autograd/ |
| D | profiler_kineto.cpp | 590 c10::get_privateuse1_backend() != "privateuseone"), in prepareProfiler()
|
| D | init.cpp | 379 if (c10::get_privateuse1_backend() != "privateuseone") { in THPAutograd_initExtension()
|
| /external/pytorch/torch/csrc/jit/serialization/ |
| D | unpickler.cpp | 630 c10::get_privateuse1_backend(), in readInstruction()
|
| /external/pytorch/torch/csrc/ |
| D | Module.cpp | 639 return THPUtils_packString(c10::get_privateuse1_backend()); in THModule_get_privateuse1_backend_name()
|
| /external/pytorch/aten/src/ATen/native/ |
| D | TensorFactories.cpp | 274 …"NYI: named tensors only support CPU, CUDA, XPU or ", c10::get_privateuse1_backend(), " tensors."); in empty_names()
|