From cf3c25b7fb635e9f75991c332deafaa4028d86dd Mon Sep 17 00:00:00 2001 From: Scott Zhu Date: Thu, 14 Mar 2019 10:41:38 -0700 Subject: [PATCH] Remove the warning message in V1 LSTM. LSTM v2 is using v1 as base class, and warning message is misleading in that case. When user starts using v2, the performance should be good since it will use cudnn kernel when available. PiperOrigin-RevId: 238473081 --- tensorflow/python/keras/layers/recurrent.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/tensorflow/python/keras/layers/recurrent.py b/tensorflow/python/keras/layers/recurrent.py index b32a8cddfa3..dccae5c6401 100644 --- a/tensorflow/python/keras/layers/recurrent.py +++ b/tensorflow/python/keras/layers/recurrent.py @@ -2508,10 +2508,6 @@ class LSTM(RNN): logging.warning('`implementation=0` has been deprecated, ' 'and now defaults to `implementation=1`.' 'Please update your layer call.') - if context.executing_eagerly() and context.num_gpus() > 0: - logging.warn('%s: Note that this layer is not optimized for performance. ' - 'Please use tf.keras.layers.CuDNNLSTM for better ' - 'performance on GPU.', self) cell = LSTMCell( units, activation=activation,