make Class — pytorch Architecture

Source Code

aten/src/ATen/core/dispatch/CppSignature.h lines 20–33

  template <class FuncType>
  static CppSignature make() {
    // Normalize functors, lambdas, function pointers, etc. into the plain
    // function type The first argument of the schema might be of type
    // DispatchKeySet, in which case we remove it. We do this to guarantee that
    // all CppSignature's for an operator will match, even if they're registered
    // with different calling conventions.
    // See Note [Plumbing Keys Through The Dispatcher]
    using decayed_function_type =
        typename c10::remove_DispatchKeySet_arg_from_func<
            std::decay_t<FuncType>>::func_type;

    return CppSignature(std::type_index(typeid(decayed_function_type)));
  }