Make out of place reducer return tensor

Author: Nikita Savelyev

src/nncf/experimental/common/tensor_statistics/collectors.py

@@ -26,6 +26,7 @@
 from nncf.quantization.advanced_parameters import AggregatorType
 from nncf.quantization.range_estimator import StatisticsType
 from nncf.tensor import Tensor
+from nncf.tensor import TensorDataType
 
 InplaceInsertionFNType = TypeVar("InplaceInsertionFNType")
 AggregationAxes = tuple[int, ...]
@@ -428,7 +429,8 @@ def __init__(self, inplace: bool = False):
         super().__init__(inplace=inplace)
 
     def _reduce_out_of_place(self, x: list[TensorType]) -> list[tuple[int, ...]]:
-        return [x[0].shape]
+        # Return as tensor for consistency
+        return [fns.tensor(x[0].shape, backend=x[0].backend, dtype=TensorDataType.int32, device=x[0].device)]
 
     def get_inplace_fn(self) -> Optional[InplaceInsertionFNType]:
         return None

src/nncf/experimental/common/tensor_statistics/statistics.py

@@ -12,7 +12,6 @@
 from __future__ import annotations
 
 from collections import Counter
-from collections import deque
 from dataclasses import dataclass
 from dataclasses import fields
 from typing import Any, ClassVar
@@ -111,7 +110,7 @@ class MeanTensorStatistic(TensorStatistic):
 
     def __post_init__(self):
         if isinstance(self.shape[0], Tensor):
-            # If in-place shape reducer and Noop aggregator were used, shape is a sequence containing a single tensor
+            # If shape reducer and Noop aggregator were used, shape is a sequence containing a single tensor
             self.shape = tuple(self.shape[0].data.tolist())
 
     def __eq__(self, other: TensorStatistic):