From 01e60d057da7e64c1fcf255e62a259f34a5f7ba0 Mon Sep 17 00:00:00 2001 From: Evgenya Stepyreva Date: Thu, 4 Jun 2020 14:18:33 +0300 Subject: [PATCH] [ MO ] InterpolateConcat empty sources fix (#764) --- model-optimizer/extensions/back/InterpolateReshape.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model-optimizer/extensions/back/InterpolateReshape.py b/model-optimizer/extensions/back/InterpolateReshape.py index e1ecbeb..11c1f28 100644 --- a/model-optimizer/extensions/back/InterpolateReshape.py +++ b/model-optimizer/extensions/back/InterpolateReshape.py @@ -72,7 +72,7 @@ class InterpolateConcat(BackReplacementPattern): if concat_axis in interp_axes: return - concat_srcs = [port.get_source() for port in concat.in_ports().values()] + concat_srcs = [port.get_source() for port in concat.in_ports().values() if not port.disconnected()] non_interp_concat_srcs = [src for src in concat_srcs if src.node.soft_get('type') != 'Interpolate'] if len(non_interp_concat_srcs) == 0: return -- 2.7.4