Home / Class/ KernelWithConstructorArg Class — pytorch Architecture

KernelWithConstructorArg Class — pytorch Architecture

Architecture documentation for the KernelWithConstructorArg class in make_boxed_from_unboxed_functor_test.cpp from the pytorch codebase.

Class cpp

Entity Profile

Source Code

aten/src/ATen/core/boxing/impl/make_boxed_from_unboxed_functor_test.cpp lines 567–578

class KernelWithConstructorArg final : public OperatorKernel {
public:
  explicit KernelWithConstructorArg(int64_t offset)
  : offset_(offset) {}

  int64_t operator()(const Tensor&, int64_t input) {
    return input + offset_;
  }

private:
  int64_t offset_;
};

Source

Analyze Your Own Codebase

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

Try Supermodel Free