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
This commit is contained in:
Scott Zhu 2019-03-14 10:41:38 -07:00 committed by TensorFlower Gardener
parent 425d4f2089
commit cf3c25b7fb

View File

@ -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,