ONE - On-device Neural Engine
Loading...
Searching...
No Matches
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
23{
24namespace odc
25{
26
27class Quantizer : public IQuantizer
28{
29public:
30 Quantizer() = default;
31 ~Quantizer() = default;
32
33 int quantize(const char *in, const char *out, QuantizeType qtype) override;
34
38 void setMinMaxRecordsThreshold(uint32_t value) { _minmax_threshold = value; };
39
46 bool readyForQuantize() override;
47
53 bool deleteMinMaxFile() override;
54
55private:
56 uint32_t _minmax_threshold = 0;
57};
58
59} // namespace odc
60} // namespace onert
61
62#endif // __ONERT_ODC_QUANTIZE_H__
void setMinMaxRecordsThreshold(uint32_t value)
Set the number of minmax records enough for quantization.
Definition Quantizer.h:38
int quantize(const char *in, const char *out, QuantizeType qtype) override
Definition Quantizer.cc:87
bool deleteMinMaxFile() override
Delete minmax file.
Definition Quantizer.cc:174
bool readyForQuantize() override
Checking the number of minmax records enough for quantization (comparison with threshold)
Definition Quantizer.cc:161