Publishing 2019 R2 content (#223)

model-optimizer/extensions/ops/interpolate.py

@@ -0,0 +1,73 @@
+"""
+ Copyright (c) 2019 Intel Corporation
+
+ Licensed under the Apache License, Version 2.0 (the "License");
+ you may not use this file except in compliance with the License.
+ You may obtain a copy of the License at
+
+      http://www.apache.org/licenses/LICENSE-2.0
+
+ Unless required by applicable law or agreed to in writing, software
+ distributed under the License is distributed on an "AS IS" BASIS,
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ See the License for the specific language governing permissions and
+ limitations under the License.
+"""
+
+from mo.front.common.layout import get_batch_dim, get_features_dim, shape_for_layout
+from mo.graph.graph import Node, Graph
+from mo.ops.op import Op, PermuteAttrs
+
+
+class Interpolate(Op):
+    op = 'Interpolate'
+    enabled = False
+
+    def __init__(self, graph: Graph, attrs: dict):
+        mandatory_props = {
+            'type': __class__.op,
+            'op': __class__.op,
+            'axes': None,
+            'mode': None,
+            'align_corners': 0,
+            'antialias': 0,
+            'pads_begin': 0,
+            'pads_end': 0,
+            'in_ports_count': 2,
+            'out_ports_count': 1,
+            'force_precision_in_ports': {1:'int64'},
+            'infer': __class__.infer,
+        }
+        super().__init__(graph, mandatory_props, attrs)
+
+    def supported_attrs(self):
+        return [
+            ('axes', lambda node: ','.join(map(str, node.axes))),
+            'mode', 'align_corners', 'antialias', 'pads_begin', 'pads_end',
+        ]
+
+    @staticmethod
+    def infer(node: Node):
+        layout = node.graph.graph['layout']
+
+        assert len(layout) == 4
+        assert len([p for p in node.in_ports().values() if not p.disconnected()])
+        assert node.has_valid('mode')
+        assert node.has_valid('axes')
+
+        src_shape = node.in_port(0).data.get_shape()
+        assert src_shape is not None
+        dst_shape = node.in_port(1).data.get_value()
+        assert dst_shape is not None
+
+        out_height = dst_shape[0]
+        out_width = dst_shape[1]
+
+        node.out_node().shape = shape_for_layout(layout,
+                                                 batch=src_shape[get_batch_dim(layout, 4)],
+                                                 features=src_shape[get_features_dim(layout, 4)],
+                                                 height=out_height,
+                                                 width=out_width)
+
+        PermuteAttrs.create_permute_attrs(node, attrs=[('axes', 'input:0')])
+