File tree 1 file changed +4
-4
lines changed
1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change 202
202
203
203
204
204
def get_num_quantized_nodes (model ):
205
- num_fake_quantize = 0
205
+ num_fake_nodes = 0
206
206
types_map = {
207
207
"i8" : "int8" ,
208
208
"u8" : "int8" ,
@@ -218,15 +218,15 @@ def get_num_quantized_nodes(model):
218
218
ov_model = model if isinstance (model , ov .Model ) else model .model
219
219
for elem in ov_model .get_ops ():
220
220
if "FakeQuantize" in elem .name :
221
- num_fake_quantize += 1
221
+ num_fake_nodes += 1
222
222
elif "FakeConvert" in elem .name :
223
- num_fake_quantize += 1
223
+ num_fake_nodes += 1
224
224
for i in range (elem .get_output_size ()):
225
225
type_name = elem .get_output_element_type (i ).get_type_name ()
226
226
if type_name in types_map :
227
227
name = types_map [type_name ]
228
228
num_weight_nodes [name ] += 1
229
- return num_fake_quantize , num_weight_nodes
229
+ return num_fake_nodes , num_weight_nodes
230
230
231
231
232
232
@contextmanager
You can’t perform that action at this time.
0 commit comments