Skip to content

Commit 7009c42

Browse files
committed
updateZZ
1 parent b45d945 commit 7009c42

File tree

2 files changed

+2
-6
lines changed

2 files changed

+2
-6
lines changed

nncf/common/graph/layer_attributes.py

-4
Original file line numberDiff line numberDiff line change
@@ -65,10 +65,6 @@ def __init__(self, weight_requires_grad: bool, dtype: Dtype = Dtype.FLOAT, with_
6565
self.dtype = dtype
6666
self.with_bias = with_bias
6767

68-
def get_num_filters(self) -> int:
69-
weight_shape = self.get_weight_shape()
70-
return weight_shape[self.get_target_dim_for_compression()]
71-
7268

7369
class GenericWeightedLayerAttributes(WeightedLayerAttributes):
7470
"""

nncf/common/graph/utils.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -204,5 +204,5 @@ def get_num_filters_legacy(layer_attributes: WeightedLayerAttributes) -> int:
204204
:param layer_attributes: layer attributes of NNCFNode.
205205
:return: number of filters.
206206
"""
207-
weight_shape = layer_attributes.get_weight_shape_legacy()
208-
return weight_shape[layer_attributes.get_target_dim_for_compression_legacy()]
207+
weight_shape = get_weight_shape_legacy(layer_attributes)
208+
return weight_shape[get_target_dim_for_compression_legacy(layer_attributes)]

0 commit comments

Comments
 (0)