Searched defs:OperatorKernel (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/boxing/ |
| 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 | BoxedKernel_impl.h | 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
|
| H A D | OperatorKernel.h | 23 struct TORCH_API OperatorKernel : public c10::intrusive_ptr_target { struct
|