Disable the new codepath of LSTM/GRU.

PiperOrigin-RevId: 337966066 Change-Id: I87b9533acdc04342d317196a77323016ee96352e
2020-10-19 17:10:34 -07:00 · 2020-10-19 17:10:34 -07:00 · 73b709743a
commit 73b709743a
parent d1f578cf6e
1 changed files with 1 additions and 1 deletions
--- a/tensorflow/python/keras/layers/recurrent_v2.py
+++ b/tensorflow/python/keras/layers/recurrent_v2.py
@ -67,7 +67,7 @@ _CUDNN_NOT_AVAILABLE_MSG = ('Layer %s will not use cuDNN kernel since it '


 def _use_new_code():
-  return True
+  return False


 # TODO(b/169707691): The wrapper can be removed if TFLite doesn't need to rely