Skip to content

Commit 71c98bd

Browse files
committed
fix checks
1 parent ff6c7b2 commit 71c98bd

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

tests/test_backend.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2933,6 +2933,7 @@ def func(input_holder):
29332933
for input_val in input_vals:
29342934
self._run_test_case(func, [_OUTPUT], {_INPUT: input_val})
29352935

2936+
@skip_tf_cpu("only tf_gpu can run CudnnGPU")
29362937
@check_opset_min_version(11, "GRU")
29372938
def test_cudnngru(self):
29382939
def func():

tf2onnx/onnx_opset/rnn.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -248,4 +248,4 @@ def NM(nm):
248248
ctx.make_node('Concat', [XNF, XNB], outputs=[node.output[0]], attr={'axis': -1})
249249
else:
250250
ctx.make_node('Identity', [XNF], outputs=[node.output[0]])
251-
ctx.make_node('Concat', YHS, outputs=[node.output[1]], attr={'axis': 0})
251+
ctx.make_node('Concat', YHS, outputs=[node.output[1]], attr={'axis': 0})

0 commit comments

Comments
 (0)