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

Fixed input cut for case when port is not specified. (#10134)

Browse Source
This commit is contained in:
Anastasia Popova 2022-02-04 19:03:12 +03:00 committed by GitHub
parent b3a990b0a7
commit 12a310636d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tools/mo/openvino/tools/mo/front/extractor.py
View File

@ -936,7 +936,7 @@ def add_input_op(graph: Graph, node_id: str, port: int = 0, data: bool = False,
tensor_name = str(port) + ":" + Node(graph, node_id).soft_get('name') tensor_name = str(port) + ":" + Node(graph, node_id).soft_get('name')
fw_info = [(Node(graph, node_id).soft_get('name'), tensor_name)] fw_info = [(Node(graph, node_id).soft_get('name'), tensor_name)]
if is_out_port and port == 0: if not is_out_port and port == 0:
tensor_name_no_port = Node(graph, node_id).soft_get('name') tensor_name_no_port = Node(graph, node_id).soft_get('name')
if graph.has_tensor_name(tensor_name_no_port): if graph.has_tensor_name(tensor_name_no_port):
log.warning('Could not add user defined input name {} to tensor names list of as ' log.warning('Could not add user defined input name {} to tensor names list of as '