Update elastic_width.py

Author: shumaari

nncf/experimental/torch/nas/bootstrapNAS/elasticity/elastic_width.py

@@ -1204,15 +1204,15 @@ def _create_dynamic_dw_conv_input_op(conv_layer_attrs: BaseLayerAttributes, node
     def _create_dynamic_bn_input_op(generic_layer_attrs: BaseLayerAttributes, node_name: str) -> UpdateBatchNormParams:
         assert isinstance(generic_layer_attrs, GenericWeightedLayerAttributes)
         dynamic_bn_input_op = ElasticInputWidthBatchNormOp(
-            max_width=generic_layer_attrs.get_num_filters(), node_name=node_name
+            max_width=generic_layer_attrs.get_num_filters_legacy(), node_name=node_name
         )
         return UpdateBatchNormParams(dynamic_bn_input_op)
 
     @staticmethod
     def _create_dynamic_ln_input_op(generic_layer_attrs: BaseLayerAttributes, node_name: str) -> UpdateLayerNormParams:
         assert isinstance(generic_layer_attrs, GenericWeightedLayerAttributes)
         dynamic_ln_input_op = ElasticInputWidthLayerNormOp(
-            max_width=generic_layer_attrs.get_num_filters(), node_name=node_name
+            max_width=generic_layer_attrs.get_num_filters_legacy(), node_name=node_name
         )
         return UpdateLayerNormParams(dynamic_ln_input_op)