pytorch Architecture Docs
Home / Class/ Hardtanh_ Class — pytorch Architecture

Hardtanh_ Class — pytorch Architecture

Architecture documentation for the Hardtanh_ class in vulkan_api_test.cpp from the pytorch codebase.

Class cpp

Entity Profile

Source Code

aten/src/ATen/test/vulkan_api_test.cpp lines 7238–7249

class Hardtanh_ final : public BaseOp {
 public:
  Hardtanh_() : BaseOp(OpType::hardtanh_) {}

  at::Tensor run(at::Tensor& input) const override {
    return at::hardtanh_(input, 0, 6);
  }

  std::string toString() const override {
    return "hardtanh_";
  }
};

Source

Analyze Your Own Codebase

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

Try Supermodel Free