重命名 pt2tf 为 pt2pb

2020-10-14 08:55:07 +08:00
parent 324ab60a5d
commit 90ae190559
407 changed files with 0 additions and 0 deletions
@@ -0,0 +1,38 @@
+import tensorflow as tf
+
+from onnx_tf.common import exception
+from onnx_tf.handlers.backend_handler import BackendHandler
+from onnx_tf.handlers.handler import onnx_op
+from onnx_tf.handlers.handler import partial_support
+from onnx_tf.handlers.handler import ps_description
+from .math_mixin import ArithmeticMixin
+
+
+@onnx_op("Mod")
+@partial_support(True)
+@ps_description("Mod Dividend or Divisor in " +
+                "int8/int16/uint8/uint16/uint32/uint64 " +
+                "are not supported in Tensorflow.")
+class Mod(ArithmeticMixin, BackendHandler):
+
+  @classmethod
+  def args_check(cls, node, **kwargs):
+    unsupported_dtype = [
+        tf.int8, tf.int16, tf.uint8, tf.uint16, tf.uint32, tf.uint64
+    ]
+    x = kwargs["tensor_dict"][node.inputs[0]]
+    y = kwargs["tensor_dict"][node.inputs[1]]
+    if x.dtype in unsupported_dtype:
+      exception.OP_UNSUPPORTED_EXCEPT(
+          "Mod Dividend in " + str(x.dtype), "Tensorflow")
+    if y.dtype in unsupported_dtype:
+      exception.OP_UNSUPPORTED_EXCEPT(
+          "Mod Divisor in " + str(y.dtype), "Tensorflow")
+
+  @classmethod
+  def version_10(cls, node, **kwargs):
+    fmod = node.attrs.get("fmod", 0)
+    tf_func = tf.floormod
+    if fmod == 1:
+      tf_func = tf.truncatemod
+    return [cls.make_tensor_from_onnx_node(node, tf_func=tf_func, **kwargs)]