replace node.get_attr('xxx').i by node.get_attr_value('xxx')

Author: lei-Qiao

tf2onnx/onnx_opset/math.py

@@ -251,11 +251,10 @@ def version_1(cls, ctx, node, **kwargs):
         # ONNX: Each input value is divided by (bias+(alpha/size)*sum(xi^2 for every xi in the local region))^beta
         # TF: sqr_sum[a, b, c, d] = sum(input[a, b, c, d - depth_radius : d + depth_radius + 1] ** 2)
         #     output = input / (bias + alpha * sqr_sum) ** beta
-        depth_radius = node.get_attr("depth_radius")
-        if depth_radius:
-            size = depth_radius.i
-        else:
-            size = 5
+
+        # by default, depth_radius is 5 in tensorflow
+        size = node.get_attr_value("depth_radius", 5) * 2 + 1
+
         node.set_attr("size", size)
         node.set_attr("alpha", size * node.get_attr("alpha").f)
 

tf2onnx/onnx_opset/tensor.py

@@ -1257,7 +1257,7 @@ def version_10(cls, ctx, node, **kwargs):
         seq_dim = node.get_attr("seq_dim")
         utils.make_sure(seq_dim is not None, "sequence dim must be given in {}".format(node.name))
         seq_dim = seq_dim.i
-        batch_dim = node.get_attr_value("batch_dim")
+        batch_dim = node.get_attr_value("batch_dim", 0)
 
         ctx.remove_node(node.name)
         node = ctx.make_node(