Home
last modified time | relevance | path

Searched refs:OperatorKernel (Results 1 – 17 of 17) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/core/boxing/impl/
H A Dmake_boxed_from_unboxed_functor_test.cpp12 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 Dmake_boxed_from_unboxed_functor.h458 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 DWrapFunctionIntoFunctor.h10 …<FuncPtr, ReturnType, guts::typelist::typelist<Parameters...>> final : public c10::OperatorKernel {
H A DWrapFunctionIntoRuntimeFunctor.h11 …FuncType, ReturnType, guts::typelist::typelist<Parameters...>> final : public c10::OperatorKernel {
/aosp_15_r20/external/pytorch/aten/src/ATen/core/boxing/
H A DBoxedKernel.h20 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 DKernelFunction_impl.h17 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 DBoxedKernel_impl.h10 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 DKernelFunction_test.cpp11 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 DKernelFunction.cpp13 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 DKernelFunction.h16 struct OperatorKernel;
175 static KernelFunction makeFromUnboxedFunctor(std::unique_ptr<OperatorKernel> kernelFunctor);
244 std::unique_ptr<OperatorKernel> functor,
H A DOperatorKernel.h23 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 Dop_registration.h162 …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 Dop_registration_test.cpp24 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 Dkernel_holder.h53 class AOTIPythonKernelHolder : public c10::OperatorKernel {
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A DChannelShuffle.cpp114 class QChannelShuffle final : public c10::OperatorKernel {
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A DVariableTypeUtils.h398 class WrapperFunctor final : public c10::OperatorKernel {
/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dpython_dispatch.cpp106 class PythonKernelHolder : public c10::OperatorKernel {