From 6673a90659d1506c98ba92ddded4b55573090667 Mon Sep 17 00:00:00 2001 From: Ivan Tikhonov Date: Thu, 29 Jul 2021 13:49:12 +0300 Subject: [PATCH] MO DeformableConv: fix a set of attributes (#6806) * MO DeformableConv: fix a set of attributes * codestyle * resolve review comments * resolve review comments --- model-optimizer/mo/ops/deformable_convolution.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/model-optimizer/mo/ops/deformable_convolution.py b/model-optimizer/mo/ops/deformable_convolution.py index 5eb573fb2147f1..14c7a39e03ef0d 100644 --- a/model-optimizer/mo/ops/deformable_convolution.py +++ b/model-optimizer/mo/ops/deformable_convolution.py @@ -27,4 +27,7 @@ def __init__(self, graph: Graph, attrs: dict): def backend_attrs(self): # the same attributes as in a regular convolution and additional attributes 'deformable_group', 'group' # and 'bilinear_interpolation_pad' - return Convolution(self.graph, {}).backend_attrs() + ['deformable_group', 'group', 'bilinear_interpolation_pad'] + attrs = Convolution(self.graph, {}).backend_attrs() + ['deformable_group', 'group'] + if self.get_opset() == 'opset8': + attrs.append('bilinear_interpolation_pad') + return attrs