Skip to content

GitLab

"googlemock/vscode:/vscode.git/clone" did not exist on "77d6b173380332b1c1bc540532641f410ec82d65"
Commit e9ac22c0 authored by qinyiqun's avatar qinyiqun
Browse files

Issue/1030: add a function to get quantization ptr

parent 21c6af2d
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
include/infinicore/nn/linear.hpp
View file @ e9ac22c0
...@@ -35,6 +35,8 @@ public: ...@@ -35,6 +35,8 @@ public:
Tensor weight_scale() const { return weight_scale_; } Tensor weight_scale() const { return weight_scale_; }
Tensor weight_zeros() const { return weight_zeros_; } Tensor weight_zeros() const { return weight_zeros_; }
std::shared_ptr<infinicore::quantization::BaseQuantization> get_quantization() const { return quantization_; }
protected: protected:
// Parameters // Parameters
INFINICORE_NN_PARAMETER(weight); INFINICORE_NN_PARAMETER(weight);
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment