From 73b709743a2eba2c912351e8d3334ef25e174c4b Mon Sep 17 00:00:00 2001 From: Yanhui Liang Date: Mon, 19 Oct 2020 17:10:34 -0700 Subject: [PATCH] Disable the new codepath of LSTM/GRU. PiperOrigin-RevId: 337966066 Change-Id: I87b9533acdc04342d317196a77323016ee96352e --- tensorflow/python/keras/layers/recurrent_v2.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow/python/keras/layers/recurrent_v2.py b/tensorflow/python/keras/layers/recurrent_v2.py index 8eddc14f5f8..d3c9111cd65 100644 --- 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