From 3ac85232cf17665c310acf2e4bb656db15315fb0 Mon Sep 17 00:00:00 2001 From: Alexander Dokuchaev Date: Mon, 2 Oct 2023 19:45:05 +0300 Subject: [PATCH] update import mean_per_channel --- .../quantization/algorithms/fast_bias_correction/algorithm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/nncf/quantization/algorithms/fast_bias_correction/algorithm.py b/nncf/quantization/algorithms/fast_bias_correction/algorithm.py index 78a05fc26bf..3ca368671aa 100644 --- a/nncf/quantization/algorithms/fast_bias_correction/algorithm.py +++ b/nncf/quantization/algorithms/fast_bias_correction/algorithm.py @@ -26,7 +26,7 @@ from nncf.common.tensor_statistics.statistic_point import StatisticPointsContainer from nncf.common.utils.backend import BackendType from nncf.common.utils.backend import get_backend -from nncf.experimental.common.tensor_statistics import statistical_functions as s_fns +from nncf.experimental.common.tensor_statistics.statistical_functions import mean_per_channel from nncf.experimental.tensor import Tensor from nncf.experimental.tensor import functions as fns from nncf.quantization.algorithms.algorithm import Algorithm @@ -318,7 +318,7 @@ def _get_bias_shift( engine = EngineFactory.create(model) raw_output = engine.infer(input_blob) q_outputs = self._backend_entity.process_model_output(raw_output, output_name) - q_outputs = s_fns.mean_per_channel(q_outputs, channel_axis) + q_outputs = mean_per_channel(q_outputs, channel_axis) bias_shift = fns.stack(output_fp) - q_outputs return bias_shift