Skip to content

Commit d31ffe8

Browse files
committed
batch_dim is set by default 0 in tf, don't need to check
1 parent 4996b10 commit d31ffe8

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

tf2onnx/onnx_opset/tensor.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1258,10 +1258,8 @@ def version_10(cls, ctx, node, **kwargs):
12581258
utils.make_sure(seq_dim is not None, "sequence dim must be given in {}".format(node.name))
12591259
seq_dim = seq_dim.i
12601260
batch_dim = node.get_attr("batch_dim")
1261-
if batch_dim is not None:
1262-
batch_dim = batch_dim.i
1263-
else:
1264-
batch_dim = 0
1261+
#batch_dim is set by default 0 in tf
1262+
batch_dim = batch_dim.i
12651263

12661264
ctx.remove_node(node.name)
12671265
node = ctx.make_node(

0 commit comments

Comments
 (0)