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

Author: lei-Qiao

tf2onnx/onnx_opset/math.py

@@ -251,8 +251,11 @@ 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
-        size = node.get_attr_value("depth_radius") * 2 + 1
-
+        depth_radius = node.get_attr("depth_radius")
+        if depth_radius:
+            size = depth_radius.i
+        else:
+            size = 5
         node.set_attr("size", size)
         node.set_attr("alpha", size * node.get_attr("alpha").f)