Skip to content
This repository has been archived by the owner on Oct 13, 2021. It is now read-only.

Support initial states for Bidirectional RNN #417

Merged
merged 14 commits into from
Apr 1, 2020
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Reinstating skip for test_rnn_state_passing for TF2
  • Loading branch information
cjermain committed Mar 30, 2020
commit 477a446dc88c4679d2c32c20876c6cd6b97ae7b9
1 change: 1 addition & 0 deletions tests/test_layers.py
Original file line number Diff line number Diff line change
@@ -1614,6 +1614,7 @@ def test_Bidirectional_seqlen_none(self):
expected = model.predict(x)
self.assertTrue(run_onnx_runtime(onnx_model.graph.name, onnx_model, x, expected, self.model_files))

@unittest.skipIf(is_tf2, 'TODO')
def test_rnn_state_passing(self):
for rnn_class in [SimpleRNN, GRU, LSTM]:
input1 = Input(shape=(None, 5))