func Class — pytorch Architecture

Source Code

aten/src/ATen/core/boxing/BoxedKernel_impl.h lines 12–21

template <BoxedKernel::BoxedKernelFunction* func>
inline void BoxedKernel::make_boxed_function(
    OperatorKernel* /*unused*/,
    const OperatorHandle& opHandle,
    DispatchKeySet /*unused*/,
    Stack* stack) {
  // Note that we're dropping the DispatchKeySet argument.
  // See Note [Plumbing Keys Through The Dispatcher 2] for details.
  func(opHandle, stack);
}