1 /*
2  * Copyright (c) 2022 Huawei Device Co., Ltd.
3  * Licensed under the Apache License, Version 2.0 (the "License");
4  * you may not use this file except in compliance with the License.
5  * You may obtain a copy of the License at
6  *
7  *     http://www.apache.org/licenses/LICENSE-2.0
8  *
9  * Unless required by applicable law or agreed to in writing, software
10  * distributed under the License is distributed on an "AS IS" BASIS,
11  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12  * See the License for the specific language governing permissions and
13  * limitations under the License.
14  */
15 
16 #ifndef NEURAL_NETWORK_RUNTIME_QUANT_PARAMS_H
17 #define NEURAL_NETWORK_RUNTIME_QUANT_PARAMS_H
18 
19 #include <vector>
20 
21 #include "cpp_type.h"
22 #include "interfaces/kits/c/neural_network_runtime/neural_network_runtime_type.h"
23 
24 namespace OHOS {
25 namespace NeuralNetworkRuntime {
26 class QuantParams {
27 public:
28     QuantParams() = default;
29     ~QuantParams() = default;
30 
31     void SetScales(const std::vector<double>& scales);
32     void SetZeroPoints(const std::vector<int32_t>& zeroPoints);
33     void SetNumBits(const std::vector<uint32_t>& numBits);
34 
35     std::vector<double> GetScales() const;
36     std::vector<int32_t> GetZeroPoints() const;
37     std::vector<uint32_t> GetNumBits() const;
38 
39     OH_NN_ReturnCode CopyToCompat(std::vector<OHOS::NeuralNetworkRuntime::QuantParam>& compatQuantParams) const;
40 
41 private:
42     std::vector<double> m_scales;
43     std::vector<int32_t> m_zeroPoints;
44     std::vector<uint32_t> m_numBits;
45 };
46 } // namespace NeuralNetworkRuntime
47 } // namespace OHOS
48 #endif // NEURAL_NETWORK_RUNTIME_QUANT_PARAMS_H