Added cast to float before log

Signed-off-by: Tom Wildenhain <tomwi@microsoft.com>
onnx · Oct 5, 2020 · ee3b48e · ee3b48e
1 parent d5e56d5
commit ee3b48e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tf2onnx/onnx_opset/math.py b/tf2onnx/onnx_opset/math.py
@@ -48,7 +48,10 @@ def version_1(cls, ctx, node, **kwargs):
 
     @classmethod
     def version_6(cls, ctx, node, **kwargs):
-        pass
+        if node.type == "Log":
+            # ORT doesn't implement Log on doubles
+            double_to_float = {onnx_pb.TensorProto.DOUBLE: onnx_pb.TensorProto.FLOAT}
+            cast_node = node.maybe_cast_input([[onnx_pb.TensorProto.FLOAT]], double_to_float)
 
 
 @tf_op(["Acos", "Asin", "Atan", "Cos", "Sin", "Tan"])