diff --git a/tf2onnx/onnx_opset/nn.py b/tf2onnx/onnx_opset/nn.py index 1721ba254..8e6c1f827 100644 --- a/tf2onnx/onnx_opset/nn.py +++ b/tf2onnx/onnx_opset/nn.py @@ -900,6 +900,8 @@ def version_11(cls, ctx, node, **kwargs): concat_shape.output[0] ] transformation_mode = "asymmetric" + if "align_corners" in node.attr and node.attr["align_corners"].i: + transformation_mode = "align_corners" if "half_pixel_centers" in node.attr and node.attr["half_pixel_centers"].i: transformation_mode = "half_pixel" resize = ctx.make_node("Resize", resize_inputs,