Unverified Commit 5fc85c8b authored by thatPepe's avatar thatPepe Committed by GitHub
Browse files

Merge pull request #1072 from InfiniTensor/issue/1030/1

 Issue/1030: add a function to get quantization ptr
parents 17777b08 e9ac22c0
......@@ -35,6 +35,8 @@ public:
Tensor weight_scale() const { return weight_scale_; }
Tensor weight_zeros() const { return weight_zeros_; }
std::shared_ptr<infinicore::quantization::BaseQuantization> get_quantization() const { return quantization_; }
protected:
// Parameters
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