Skip to content

Commit 003f230

Browse files
committed
GRUBlockCell should be imported
registeration work is done when importing, otherwise when we restore graph from pb file and the graph contains GRUBlockCell then it will fail to run.
1 parent 5081424 commit 003f230

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/run_pretrained_models.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@
3030
from tf2onnx import utils
3131
from tf2onnx.graph import GraphUtil
3232
from tf2onnx.tfonnx import process_tf_graph
33+
from tensorflow.contrib.rnn import GRUBlockCell # pylint: disable=unused-import
3334

3435
# pylint: disable=broad-except,logging-not-lazy,unused-argument,unnecessary-lambda
3536

0 commit comments

Comments
 (0)