diff options
Diffstat (limited to 'model-optimizer/extensions/front/caffe/interp_ext.py')
| -rw-r--r-- | model-optimizer/extensions/front/caffe/interp_ext.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/model-optimizer/extensions/front/caffe/interp_ext.py b/model-optimizer/extensions/front/caffe/interp_ext.py index b1d5ba733..ae8a8da55 100644 --- a/model-optimizer/extensions/front/caffe/interp_ext.py +++ b/model-optimizer/extensions/front/caffe/interp_ext.py @@ -39,6 +39,9 @@ class InterpFrontExtractor(FrontExtractorOp): mapping_rule = merge_attrs(param, update_attrs) + # in Caffe can be 2 inputs, shape should be got from shape of the second input + mapping_rule['parse_2nd_input'] = 'shape' + # update the attributes of the node Op.get_op_class_by_name(__class__.op).update_node_stat(node, mapping_rule) return __class__.enabled |