callUnboxedKernelFunction Class — pytorch Architecture

Architecture documentation for the callUnboxedKernelFunction class in KernelFunction_impl.h from the pytorch codebase.

Class c

Entity Profile

Source Code

aten/src/ATen/core/boxing/KernelFunction_impl.h lines 95–105

template <class Return, class... Args>
inline Return callUnboxedKernelFunction(
    void* unboxed_kernel_func,
    OperatorKernel* functor,
    DispatchKeySet dispatchKeySet,
    Args&&... args) {
  using ActualSignature = Return(OperatorKernel*, DispatchKeySet, Args...);
  ActualSignature* func =
      reinterpret_cast<ActualSignature*>(unboxed_kernel_func);
  return (*func)(functor, dispatchKeySet, std::forward<Args>(args)...);
}

Source

View on GitHub

Analyze Your Own Codebase

Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.

Try Supermodel Free