Skip to content

Commit

Permalink
upd to RawReducer
Browse files Browse the repository at this point in the history
  • Loading branch information
kshpv committed Nov 14, 2024
1 parent 9d15d25 commit fb2e858
Showing 1 changed file with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@
from nncf.experimental.common.tensor_statistics.collectors import MeanAbsMaxReducer
from nncf.experimental.common.tensor_statistics.collectors import MeanAggregator
from nncf.experimental.common.tensor_statistics.collectors import MeanVarianceReducer
from nncf.experimental.common.tensor_statistics.collectors import NoopReducer
from nncf.experimental.common.tensor_statistics.collectors import RawReducer
from nncf.experimental.common.tensor_statistics.collectors import TensorCollector


Expand Down Expand Up @@ -58,7 +58,7 @@ def check_reducer(self, collector: TensorCollector, expected_reducer_type):
@pytest.mark.parametrize(
"algo_func, aggregator_type, reducer_type",
[
("get_hawq_with_backend", HAWQAggregator, NoopReducer),
("get_hawq_with_backend", HAWQAggregator, RawReducer),
("get_mean_variance_with_backend", MeanAggregator, MeanVarianceReducer),
("get_max_variance_with_backend", MeanAggregator, MaxVarianceReducer),
("get_mean_max_with_backend", MeanAggregator, MeanAbsMaxReducer),
Expand Down

0 comments on commit fb2e858

Please sign in to comment.