opset 12 support

Author: RandySheriffH

tests/test_backend.py

@@ -3130,6 +3130,29 @@ def func(X, K):
                 k_val = np.array(raw_k).astype(np.int32)
                 self._run_test_case(func, [_OUTPUT, _OUTPUT1], {_INPUT: x_val, _INPUT1: k_val})
 
+    @check_opset_min_version(12)
+    def test_inverse(self):
+        x_val = np.random.random([5, 5]).astype(np.float32)
+        def func(x):
+            return tf.linalg.inv(x, name=_TFOUTPUT)
+        self._run_test_case(func, [_OUTPUT], {_INPUT: x_val})
+
+    @check_opset_min_version(12)
+    def test_less_or_equal(self):
+        x_val = np.random.random([4, 5]).astype(np.float32)
+        y_val = np.random.random([4, 5]).astype(np.float32)
+        def func(x, y):
+            return tf.math.less_equal(x, y, name=_TFOUTPUT)
+        self._run_test_case(func, [_OUTPUT], {_INPUT: x_val, _INPUT1: y_val})
+
+    @check_opset_min_version(12)
+    def test_squared_distance(self):
+        x_val = np.random.random([4, 5]).astype(np.float32)
+        y_val = np.random.random([4, 5]).astype(np.float32)
+        def func(x, y):
+            return tf.math.squared_difference(x, y, name=_TFOUTPUT)
+        self._run_test_case(func, [_OUTPUT], {_INPUT: x_val, _INPUT1: y_val})
+
 
 if __name__ == '__main__':
     unittest_main()

tf2onnx/onnx_opset/math.py

@@ -22,7 +22,7 @@
 
 # pylint: disable=unused-argument,missing-docstring
 
-@tf_op(["Add", "AddV2", "Div", "Mul", "Sub"])
+@tf_op(["Add", "AddV2", "Div", "Mul", "Sub", "LessOrEqual"])
 class BroadcastOp(common.BroadcastOp):
     pass
 
@@ -544,3 +544,26 @@ def version_11(cls, ctx, node, **kwargs):
             cast_back_node.set_attr("to", dtypes[0])
             ctx.set_dtype(cast_back_node.output[0], dtypes[0])
             ctx.copy_shape(node.name, cast_back_node.output[0])
+
+@tf_op("MatrixInverse")
+class Inverse:
+
+    @classmethod
+    def version_12(cls, ctx, node, **kwargs):
+        utils.make_sure(node.get_attr('adjoint').i == 0, "adjoint must be false")
+        shapes = node.output_shapes
+        dtypes = node.output_dtypes
+        ctx.remove_node(node.name)
+        ctx.make_node("Inverse", inputs=node.input, outputs=node.output, name=node.name,
+                      shapes=shapes, dtypes=dtypes)
+
+@tf_op("SquaredDistance")
+class SquaredDistance:
+
+    @classmethod
+    def version_12(cls, ctx, node, **kwargs):
+        shapes = node.output_shapes
+        dtypes = node.output_dtypes
+        ctx.remove_node(node.name)
+        ctx.make_node("MeanSquaredDistance", inputs=node.input, outputs=node.output, name=node.name,
+                      shapes=shapes, dtypes=dtypes, attr={"reduction": "none"})