IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add [rename conformance ir] input shape to hash name for Convolution (#21484)

Browse Source
This commit is contained in:
Sofya Balandina 2023-12-11 19:32:07 +00:00 committed by GitHub
parent e1b9d8c167
commit cc8aeb0c09
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/tests/test_utils/functional_test_utils/layer_tests_summary/rename_conformance_ir.py
View File

@ -106,9 +106,16 @@ def generate_node_hash(node):
str_to_hash += re.sub(r"[\s+\[\]\{\}\']", "", str(node.get_attributes()))
except:
logger.error(f"Impossible to get attributes for {node.name}")
try:
partial_shape = input.get_partial_shape()
if 'Convolution' in str(input_node.get_type_info().name):
offset = 2
if 'GroupConvolution' in str(input_node.get_type_info().name) or\
'GroupConvolutionBackpropData' in str(input_node.get_type_info().name):
offset = 3
shape_str += '[' + ','.join([str(val) for val in list(partial_shape)[offset:]]) + ']'
shape_str += str(len(partial_shape))
shape_str += str(partial_shape.is_dynamic)
except: