Home
last modified time | relevance | path

Searched defs:dispatch_key (Results 1 – 25 of 27) sorted by relevance

12

/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dtensor_new.cpp538 c10::DispatchKey dispatch_key, in check_base_legacy_new()
582 c10::DispatchKey dispatch_key, in check_legacy_ctor_device()
602 c10::DispatchKey dispatch_key, in legacy_sparse_tensor_generic_ctor_new()
694 c10::DispatchKey dispatch_key) { in typeIdWithDefault()
706 c10::DispatchKey dispatch_key, in legacy_tensor_generic_ctor_new()
825 c10::DispatchKey dispatch_key, in legacy_tensor_ctor()
835 c10::DispatchKey dispatch_key, in legacy_tensor_new()
887 c10::DispatchKey dispatch_key, in sparse_compressed_tensor_ctor_worker()
1064 c10::DispatchKey dispatch_key, in sparse_compressed_tensor_ctor()
1077 c10::DispatchKey dispatch_key, in sparse_csr_tensor_ctor()
[all …]
H A Dpython_dispatch.cpp117 c10::DispatchKey dispatch_key, in PythonKernelHolder()
649 [](c10::DispatchKey dispatch_key, bool desired_state) { in initDispatchBindings()
654 [](c10::DispatchKey dispatch_key) { in initDispatchBindings()
659 [](c10::DispatchKey dispatch_key, bool desired_state) { in initDispatchBindings()
664 [](c10::DispatchKey dispatch_key) { in initDispatchBindings()
854 [](const char* dispatch_key = "") { in initDispatchBindings()
868 [](const char* dispatch_key) -> std::optional<c10::DispatchKey> { in initDispatchBindings()
878 [](const char* dispatch_key = "") { in initDispatchBindings()
H A Dnested.cpp18 c10::DispatchKey dispatch_key) { in typeIdWithDefault()
27 c10::DispatchKey dispatch_key, in nested_tensor_ctor()
/aosp_15_r20/external/pytorch/aten/src/ATen/core/op_registration/
H A Dop_registration.h72 Options&& kernel(DispatchKey dispatch_key) && { in kernel()
161 …_if_t<guts::is_functor<KernelFunctor>::value, Options&&> kernel(DispatchKey dispatch_key, Construc… in kernel()
241 …able_if_t<guts::is_function_type<FuncType>::value, Options&&> kernel(DispatchKey dispatch_key) && { in kernel()
285 …if_t<guts::is_function_type<FuncType>::value, Options&&> kernel(DispatchKey dispatch_key, FuncType… in kernel()
334 Options&&> kernel(DispatchKey dispatch_key, Lambda&& functor) && { in kernel()
402 …Options&& kernel(std::optional<DispatchKey> dispatch_key, KernelFunction&& func, std::optional<imp… in kernel()
428 std::optional<DispatchKey> dispatch_key; member
/aosp_15_r20/external/pytorch/aten/src/ATen/core/dispatch/
H A DOperatorEntry.cpp106 std::optional<DispatchKey> dispatch_key, in registerKernel()
186 std::optional<DispatchKey> dispatch_key, in registerKernel()
206 void OperatorEntry::updateFallback(const c10::Dispatcher& dispatcher, DispatchKey dispatch_key) { in registerKernel()
390 …atorEntry::updateDispatchTableEntry_(const c10::Dispatcher& dispatcher, DispatchKey dispatch_key) { in registerKernel()
404 … OperatorEntry::updateDispatchTable_(const c10::Dispatcher& dispatcher, DispatchKey dispatch_key) { in registerKernel()
516 static std::string post_process_dispatch_key_str(std::string dispatch_key) { in registerKernel()
H A DDispatcher.cpp341 std::optional<DispatchKey> dispatch_key, in registerImpl()
373 …_(const OperatorHandle& op, const OperatorName& op_name, std::optional<DispatchKey> dispatch_key, … in deregisterImpl_()
H A DOperatorEntry.h285 std::optional<DispatchKey> dispatch_key; member
/aosp_15_r20/external/pytorch/torch/
H A Dlibrary.py75 def __init__(self, ns, kind, dispatch_key=""): argument
208 def _impl_with_aoti_compile(self, op_name, dispatch_key=""): argument
255 def impl(self, op_name, fn, dispatch_key="", *, with_keyset=False): argument
333 def fallback(self, fn, dispatch_key="", *, with_keyset=False): argument
591 def _(lib: Library, name, dispatch_key=""): argument
H A D_ops.py284 def fallthrough(self, dispatch_key): argument
289 def dispatch(self, /, dispatch_key, *args, **kwargs): argument
/aosp_15_r20/external/pytorch/aten/src/ATen/core/
H A Dlibrary.cpp185 auto dispatch_key = f.dispatch_key_.has_value() ? f.dispatch_key_ : dispatch_key_; in _def() local
238 auto dispatch_key = f.dispatch_key_.has_value() ? f.dispatch_key_ : dispatch_key_; in _impl() local
270 auto dispatch_key = f.dispatch_key_.has_value() ? f.dispatch_key_ : dispatch_key_; in _fallback() local
/aosp_15_r20/external/pytorch/aten/src/ATen/core/boxing/impl/
H A Dkernel_stackbased_test.cpp56 void expectCallsIncrement(DispatchKey dispatch_key) { in expectCallsIncrement()
60 void expectCallsIncrementUnboxed(DispatchKey dispatch_key) { in expectCallsIncrementUnboxed()
70 void expectCallsDecrement(DispatchKey dispatch_key) { in expectCallsDecrement()
H A Dkernel_function_test.cpp36 void expectCallsIncrement(DispatchKey dispatch_key) { in expectCallsIncrement()
47 void expectCallsDecrement(DispatchKey dispatch_key) { in expectCallsDecrement()
655 void expectCallsConcatUnboxed(DispatchKey dispatch_key) { in expectCallsConcatUnboxed()
H A Dkernel_lambda_test.cpp23 void expectCallsIncrement(DispatchKey dispatch_key) { in expectCallsIncrement()
34 void expectCallsDecrement(DispatchKey dispatch_key) { in expectCallsDecrement()
566 void expectCallsConcatUnboxed(DispatchKey dispatch_key) { in expectCallsConcatUnboxed()
H A Dtest_helpers.h64 inline void expectDoesntFindKernel(const char* op_name, c10::DispatchKey dispatch_key) { in expectDoesntFindKernel()
H A Dmake_boxed_from_unboxed_functor_test.cpp43 void expectCallsIncrement(DispatchKey dispatch_key) { in expectCallsIncrement()
785 void expectCallsConcatUnboxed(DispatchKey dispatch_key) { in expectCallsConcatUnboxed()
H A Dkernel_function_legacy_test.cpp43 void expectCallsIncrement(DispatchKey dispatch_key) { in expectCallsIncrement()
898 void expectCallsConcatUnboxed(DispatchKey dispatch_key) { in expectCallsConcatUnboxed()
H A Dkernel_lambda_legacy_test.cpp33 void expectCallsIncrement(DispatchKey dispatch_key) { in expectCallsIncrement()
836 void expectCallsConcatUnboxed(DispatchKey dispatch_key) { in expectCallsConcatUnboxed()
/aosp_15_r20/external/pytorch/c10/core/
H A DTensorOptions.h726 inline Layout dispatchKeyToLayout(DispatchKey dispatch_key) { in dispatchKeyToLayout()
744 inline c10::DeviceType dispatchKeyToDeviceType(DispatchKey dispatch_key) { in dispatchKeyToDeviceType()
771 inline TensorOptions dispatchKeyToTensorOptions(DispatchKey dispatch_key) { in dispatchKeyToTensorOptions()
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A Dautocast_mode.cpp10 at::DispatchKey dispatch_key = get_autocast_dispatch_key_from_device_type(device_type); in is_autocast_enabled() local
15 at::DispatchKey dispatch_key = get_autocast_dispatch_key_from_device_type(device_type); in set_autocast_enabled() local
/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dlazy_tensor_test.cpp5 void LazyTensorTest(c10::DispatchKey dispatch_key, at::DeviceType device_type) { in LazyTensorTest()
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dpython_legacy_variable.cpp68 auto dispatch_key = torch::tensors::get_default_dispatch_key(); in THPVariable_pynew() local
/aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/
H A Dts_eager_fallback.cpp101 c10::DispatchKey dispatch_key(c10::DeviceType device_type) { in dispatch_key() function
/aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_eager/
H A Dkernel_holder.cpp168 c10::DispatchKey dispatch_key, in AOTIPythonKernelHolder()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/
H A DSparseTensor.cpp154 DispatchKey dispatch_key; in new_sparse() local
H A DSparseCsrTensor.cpp354 DispatchKey dispatch_key = DispatchKey::Undefined; in new_compressed_tensor() local

12