/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 {
|