Searched refs:FeatureScaleStatMethod (Results 1 – 4 of 4) sorted by relevance
| /dports/misc/mnn/MNN-1.2.0/tools/train/source/nn/ |
| H A D | NN.hpp | 29 enum FeatureScaleStatMethod { enum in MNN::Express::NN
63 FeatureScaleStatMethod featureMethod = PerChannel,
74 FeatureScaleStatMethod featureMethod = PerChannel,
78 … NN::FeatureScaleStatMethod featureScaleStatMethod = PerTensor,
90 …static bool turnQuantize(Module* module, const int bits = 8, NN::FeatureScaleStatMethod featureSca…
|
| H A D | NN.cpp | 537 NN::FeatureScaleStatMethod featureScaleStatMethod, in ConvBNReluFusedModule()
916 NN::FeatureScaleStatMethod mFeatureScaleStatMethod;
922 NN::FeatureScaleStatMethod featureScaleStatMethod, in ConvBNReluFused()
928 …tializer> weightInit, std::shared_ptr<Initializer> biasInit, NN::FeatureScaleStatMethod featureMet… in ConvInt8()
932 Module* NN::ConvInt8(const ConvParameters& para, int bits, NN::FeatureScaleStatMethod featureMethod… in ConvInt8()
937 bool NN::turnQuantize(Module* module, const int bits, NN::FeatureScaleStatMethod featureScaleStatMe… in turnQuantize()
|
| /dports/misc/mnn/MNN-1.2.0/tools/train/source/demo/ |
| H A D | quanByMSE.cpp | 44 static NN::FeatureScaleStatMethod gFeatureScale = NN::PerChannel;
|
| /dports/misc/mnn/MNN-1.2.0/pymnn/src/ |
| H A D | MNN.cc | 3246 py::enum_<NN::FeatureScaleStatMethod>(compress_module, "Feature_Scale_Method") in MOD_INIT_FUNC()
3257 py::arg("feature_scale_method") = NN::FeatureScaleStatMethod::PerTensor, in MOD_INIT_FUNC()
|