We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e2c7806 commit 96aa600Copy full SHA for 96aa600
tools/accuracy_checker/openvino/tools/accuracy_checker/utils.py
@@ -950,6 +950,15 @@ def parse_partial_shape(partial_shape):
950
ps = str(partial_shape)
951
preprocessed = ps.replace('{', '(').replace('}', ')').replace('?', '-1')
952
if '[' not in preprocessed:
953
+ preprocessed = preprocessed.replace('(', '').replace(')', '')
954
+ if '..' in preprocessed:
955
+ shape_list = []
956
+ for dim in preprocessed.split(','):
957
+ if '..' in dim:
958
+ shape_list.append(string_to_tuple(dim.replace('..', ','), casting_type=int))
959
+ else:
960
+ shape_list.append(int(dim))
961
+ return shape_list
962
return string_to_tuple(preprocessed, casting_type=int)
963
shape_list = []
964
s_pos = 0
0 commit comments