Skip to content

Commit ac7b57a

Browse files
Correctness
1 parent 6f5cd5b commit ac7b57a

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/openvino/utils_tests.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -202,7 +202,7 @@
202202

203203

204204
def get_num_quantized_nodes(model):
205-
num_fake_quantize = 0
205+
num_fake_nodes = 0
206206
types_map = {
207207
"i8": "int8",
208208
"u8": "int8",
@@ -218,15 +218,15 @@ def get_num_quantized_nodes(model):
218218
ov_model = model if isinstance(model, ov.Model) else model.model
219219
for elem in ov_model.get_ops():
220220
if "FakeQuantize" in elem.name:
221-
num_fake_quantize += 1
221+
num_fake_nodes += 1
222222
elif "FakeConvert" in elem.name:
223-
num_fake_quantize += 1
223+
num_fake_nodes += 1
224224
for i in range(elem.get_output_size()):
225225
type_name = elem.get_output_element_type(i).get_type_name()
226226
if type_name in types_map:
227227
name = types_map[type_name]
228228
num_weight_nodes[name] += 1
229-
return num_fake_quantize, num_weight_nodes
229+
return num_fake_nodes, num_weight_nodes
230230

231231

232232
@contextmanager

0 commit comments

Comments
 (0)