From 56dab165d73b9d3c817182ed039a231246d5f1b0 Mon Sep 17 00:00:00 2001 From: Tom Wildenhain Date: Tue, 18 May 2021 14:57:03 -0400 Subject: [PATCH] Fix Conv for unknown input rank Signed-off-by: Tom Wildenhain --- tf2onnx/onnx_opset/nn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tf2onnx/onnx_opset/nn.py b/tf2onnx/onnx_opset/nn.py index 3b7da8e47..da6738ee7 100644 --- a/tf2onnx/onnx_opset/nn.py +++ b/tf2onnx/onnx_opset/nn.py @@ -349,7 +349,7 @@ def any_version(cls, opset, ctx, node, **kwargs): # prefix with batch dim of [1] to satisfy rank requirements input_shape = ctx.get_shape(node.input[0]) - if len(input_shape) == spatial + 1: + if input_shape is not None and len(input_shape) == spatial + 1: gb = GraphBuilder(ctx) usq_node = gb.make_unsqueeze({"axes": [0], 'data': node.input[0]}, return_node=True) ctx.replace_inputs(node, [usq_node.output[0]] + node.input[1:])