| /aosp_15_r20/external/pytorch/aten/src/ATen/core/boxing/impl/ |
| H A D | make_boxed_from_unboxed_functor_test.cpp | 12 using c10::OperatorKernel;
24 struct ErrorKernel final : public OperatorKernel {
31 struct IncrementKernel final : OperatorKernel {
37 struct DecrementKernel final : OperatorKernel {
78 struct KernelWithoutOutput final : OperatorKernel {
95 struct KernelWithZeroOutputs final : OperatorKernel {
113 struct KernelWithIntOutput final : OperatorKernel {
131 struct KernelWithTensorOutput final : OperatorKernel {
154 struct KernelWithTensorListOutput final : OperatorKernel {
175 struct KernelWithIntListOutput final : OperatorKernel {
[all …]
|
| H A D | make_boxed_from_unboxed_functor.h | 458 static ReturnType call(OperatorKernel* functor, DispatchKeySet, ParameterTypes... args) {
481 …static ReturnType call(OperatorKernel* functor, DispatchKeySet dispatchKeySet, ParameterTypes... a…
496 …call_functor_with_args_from_stack_(OperatorKernel* functor, DispatchKeySet dispatchKeySet, Stack* …
511 …call_functor_with_args_from_stack(OperatorKernel* functor, DispatchKeySet dispatchKeySet, Stack* s…
567 static_assert(std::is_base_of<OperatorKernel, KernelFunctor>::value,
570 …static void call(OperatorKernel* functor, const OperatorHandle&, DispatchKeySet dispatchKeySet, St…
599 using OperatorKernel = c10::OperatorKernel;
|
| H A D | WrapFunctionIntoFunctor.h | 10 …<FuncPtr, ReturnType, guts::typelist::typelist<Parameters...>> final : public c10::OperatorKernel {
|
| H A D | WrapFunctionIntoRuntimeFunctor.h | 11 …FuncType, ReturnType, guts::typelist::typelist<Parameters...>> final : public c10::OperatorKernel {
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/boxing/ |
| H A D | BoxedKernel.h | 20 TORCH_API void fallthrough_kernel(OperatorKernel*, const OperatorHandle&, DispatchKeySet, Stack*);
60 TORCH_API void ambiguous_autogradother_kernel(OperatorKernel*, const OperatorHandle&, DispatchKeySe…
69 [[noreturn]] TORCH_API void named_not_supported_kernel(OperatorKernel*, const OperatorHandle&, Disp…
101 …using InternalBoxedKernelFunction = void(OperatorKernel*, const OperatorHandle&, DispatchKeySet, S…
160 …static void make_boxed_function(OperatorKernel*, const OperatorHandle& opHandle, DispatchKeySet, S…
163 …static void make_boxed_function(OperatorKernel*, const OperatorHandle& opHandle, DispatchKeySet, S…
165 …explicit BoxedKernel(std::unique_ptr<OperatorKernel> functor, InternalBoxedKernelFunction* boxed_k…
167 OperatorKernel* getFunctor() const;
170 c10::intrusive_ptr<OperatorKernel> functor_;
|
| H A D | KernelFunction_impl.h | 17 inline KernelFunction::KernelFunction(std::unique_ptr<OperatorKernel> functor, InternalBoxedKernelF…
50 inline Return callUnboxedKernelFunction(void* unboxed_kernel_func, OperatorKernel* functor, Dispatc… in callUnboxedKernelFunction()
51 using ActualSignature = Return (OperatorKernel*, DispatchKeySet, Args...); in callUnboxedKernelFunction()
149 inline KernelFunction KernelFunction::makeFromUnboxedFunctor(std::unique_ptr<OperatorKernel> kernel… in makeFromUnboxedFunctor()
154 …static_assert(std::is_base_of<OperatorKernel, KernelFunctor>::value, "Tried to call KernelFunction… in makeFromUnboxedFunctor()
182 … guts::make_unique_base<OperatorKernel, typename impl::WrapFunctionIntoFunctor<FuncPtr>::type>() in makeFromUnboxedFunction()
199 …guts::make_unique_base<OperatorKernel, impl::WrapFunctionIntoRuntimeFunctor<std::decay_t<FuncType>… in makeFromUnboxedRuntimeFunction()
209 …guts::make_unique_base<OperatorKernel, impl::WrapFunctionIntoRuntimeFunctor<std::decay_t<Lambda>>>… in makeFromUnboxedLambda()
225 …guts::make_unique_base<OperatorKernel, impl::WrapFunctionIntoRuntimeFunctor<std::decay_t<Lambda>>>… in makeFromUnboxedLambda()
|
| H A D | BoxedKernel_impl.h | 10 inline BoxedKernel::BoxedKernel(std::unique_ptr<OperatorKernel> functor, InternalBoxedKernelFunctio… in BoxedKernel()
16 inline void BoxedKernel::make_boxed_function(OperatorKernel*, const OperatorHandle& opHandle, Dispa… in make_boxed_function() argument
23 inline void BoxedKernel::make_boxed_function(OperatorKernel*, const OperatorHandle& opHandle, Dispa… in make_boxed_function() argument
83 …static_assert(std::is_base_of<OperatorKernel, KernelFunctor>::value, "Tried to call BoxedKernel::m… in makeFromFunctor()
86 [](OperatorKernel* kernel, const OperatorHandle& op, DispatchKeySet ks, Stack* stack) { in makeFromFunctor()
92 inline OperatorKernel* BoxedKernel::getFunctor() const { in getFunctor()
|
| H A D | KernelFunction_test.cpp | 11 using c10::OperatorKernel;
68 struct unboxed_functor_with_return final : OperatorKernel {
75 struct unboxed_functor_without_return final : OperatorKernel {
82 std::unique_ptr<OperatorKernel> operator()() { in operator ()()
88 std::unique_ptr<OperatorKernel> operator()() { in operator ()()
446 …Functor<false, kernels::unboxed_functor_with_return>(std::unique_ptr<OperatorKernel>(std::make_uni… in TEST()
451 …ctor<false, kernels::unboxed_functor_without_return>(std::unique_ptr<OperatorKernel>(std::make_uni… in TEST()
456 …Functor<false, kernels::unboxed_functor_with_return>(std::unique_ptr<OperatorKernel>(std::make_uni… in TEST()
461 …ctor<false, kernels::unboxed_functor_without_return>(std::unique_ptr<OperatorKernel>(std::make_uni… in TEST()
|
| H A D | KernelFunction.cpp | 13 void fallthrough_kernel(OperatorKernel*, const OperatorHandle&, DispatchKeySet, Stack*) { in fallthrough_kernel() argument
22 void ambiguous_autogradother_kernel(OperatorKernel*, const OperatorHandle& op, DispatchKeySet, Stac… in ambiguous_autogradother_kernel() argument
35 void named_not_supported_kernel(OperatorKernel*, const OperatorHandle& op, DispatchKeySet, Stack*) { in named_not_supported_kernel() argument
|
| H A D | KernelFunction.h | 16 struct OperatorKernel;
175 static KernelFunction makeFromUnboxedFunctor(std::unique_ptr<OperatorKernel> kernelFunctor);
244 std::unique_ptr<OperatorKernel> functor,
|
| H A D | OperatorKernel.h | 23 struct TORCH_API OperatorKernel : public c10::intrusive_ptr_target { struct
24 ~OperatorKernel() override = default;
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/op_registration/ |
| H A D | op_registration.h | 162 …static_assert(std::is_base_of<OperatorKernel, KernelFunctor>::value, "Tried to register a kernel f… in kernel()
214 …static_assert(std::is_base_of<OperatorKernel, KernelFunctor>::value, "Tried to register a kernel f… in catchAllKernel()
335 …static_assert(!std::is_base_of<OperatorKernel, std::decay_t<Lambda>>::value, "The kernel(x) API fo… in kernel()
376 …static_assert(!std::is_base_of<OperatorKernel, std::decay_t<Lambda>>::value, "The kernel(x) API fo… in catchAllKernel()
552 …static_assert(!std::is_base_of<OperatorKernel, Lambda>::value, "c10::OperatorKernel is part of the…
569 …static_assert(!std::is_base_of<OperatorKernel, Lambda>::value, "c10::OperatorKernel is part of the…
|
| H A D | op_registration_test.cpp | 24 using c10::OperatorKernel;
37 struct DummyKernel final : OperatorKernel {
41 struct MockKernel final : OperatorKernel {
406 struct DummyKernelWithIntParam final : OperatorKernel {
773 struct ArgTypeTestKernel final : OperatorKernel {
|
| /aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_eager/ |
| H A D | kernel_holder.h | 53 class AOTIPythonKernelHolder : public c10::OperatorKernel {
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | ChannelShuffle.cpp | 114 class QChannelShuffle final : public c10::OperatorKernel {
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | VariableTypeUtils.h | 398 class WrapperFunctor final : public c10::OperatorKernel {
|
| /aosp_15_r20/external/pytorch/torch/csrc/utils/ |
| H A D | python_dispatch.cpp | 106 class PythonKernelHolder : public c10::OperatorKernel {
|