ONE - On-device Neural Engine
All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Modules Pages
Quantizer.h
Go to the documentation of this file.
1/*
2 * Copyright (c) 2023 Samsung Electronics Co., Ltd. All Rights Reserved
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17#ifndef __ONERT_ODC_QUANTIZE_H__
18#define __ONERT_ODC_QUANTIZE_H__
19
20#include "odc/IQuantizer.h"
21
22namespace onert::odc
23{
24
25class Quantizer : public IQuantizer
26{
27public:
28 Quantizer() = default;
29 ~Quantizer() = default;
30
31 int quantize(const char *in, const char *out, QuantizeType qtype) override;
32
36 void setMinMaxRecordsThreshold(uint32_t value) override { _minmax_threshold = value; };
37
44 bool readyForQuantize() override;
45
51 bool deleteMinMaxFile() override;
52
53private:
54 uint32_t _minmax_threshold = 0;
55};
56
57} // namespace onert::odc
58
59#endif // __ONERT_ODC_QUANTIZE_H__
void setMinMaxRecordsThreshold(uint32_t value) override
Set the number of minmax records enough for quantization.
Definition Quantizer.h:36
int quantize(const char *in, const char *out, QuantizeType qtype) override
Definition Quantizer.cc:85
bool deleteMinMaxFile() override
Delete minmax file.
Definition Quantizer.cc:172
bool readyForQuantize() override
Checking the number of minmax records enough for quantization (comparison with threshold)
Definition Quantizer.cc:159