Home / Class/ KernelWithMultipleConstructorArgs Class — pytorch Architecture

KernelWithMultipleConstructorArgs Class — pytorch Architecture

Architecture documentation for the KernelWithMultipleConstructorArgs 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 597–608

class KernelWithMultipleConstructorArgs final : public OperatorKernel {
public:
  explicit KernelWithMultipleConstructorArgs(int64_t offset1, int64_t offset2)
  : offset_(offset1 + offset2) {}

  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