Home
last modified time | relevance | path

Searched refs:BoxedKernelFunction (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/core/op_registration/
H A Dop_registration.h71 template<KernelFunction::BoxedKernelFunction* kernel_func>
77 template<KernelFunction::BoxedKernelFunction* kernel_func>
242 …static_assert(!std::is_same<FuncType, KernelFunction::BoxedKernelFunction>::value, "Tried to regis… in kernel()
271 …static_assert(!std::is_same<FuncType, KernelFunction::BoxedKernelFunction>::value, "Tried to regis… in catchAllKernel()
286 …static_assert(!std::is_same<FuncType, KernelFunction::BoxedKernelFunction>::value, "Tried to regis… in kernel()
301 …static_assert(!std::is_same<FuncType, KernelFunction::BoxedKernelFunction>::value, "Tried to regis… in catchAllKernel()
333 …er_function_traits_t<std::decay_t<Lambda>>::func_type, KernelFunction::BoxedKernelFunction>::value,
374 …er_function_traits_t<std::decay_t<Lambda>>::func_type, KernelFunction::BoxedKernelFunction>::value,
521 …n_type<FuncType>::value && !std::is_same<FuncType, KernelFunction::BoxedKernelFunction>::value, Re…
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DCPUFallback.h19 template<c10::KernelFunction::BoxedKernelFunction* fallback_fn, class Op, bool symint, class Return…
22 template<c10::KernelFunction::BoxedKernelFunction* fallback_fn, class Op, bool symint, class Return…
40 template<c10::KernelFunction::BoxedKernelFunction* fallback_fn, class Op>
43 template<c10::KernelFunction::BoxedKernelFunction* fallback_fn, class Op>
/aosp_15_r20/external/pytorch/aten/src/ATen/core/boxing/
H A DBoxedKernel.h103 using BoxedKernelFunction = void(const OperatorHandle&, Stack*); variable
126 template<BoxedKernelFunction* func>
159 template<BoxedKernelFunction* func>
H A DKernelFunction.h86 using BoxedKernelFunction = BoxedKernel::BoxedKernelFunction;
153 template<BoxedKernelFunction* func>
H A DKernelFunction_impl.h121 template<KernelFunction::BoxedKernelFunction* func>
176 …static_assert(!std::is_same<typename FuncPtr::FuncType, BoxedKernelFunction>::value, "Tried to cal… in makeFromUnboxedFunction()
195 …static_assert(!std::is_same<FuncType, BoxedKernelFunction>::value, "Tried to call KernelFunction::… in makeFromUnboxedRuntimeFunction()
H A DBoxedKernel_impl.h15 template<BoxedKernel::BoxedKernelFunction* func>
44 template<BoxedKernel::BoxedKernelFunction* func>
/aosp_15_r20/external/pytorch/torch/
H A Dlibrary.h244 template <c10::BoxedKernel::BoxedKernelFunction* func>