Forward compat change to LSTM.
PiperOrigin-RevId: 327904183
Change-Id: Ib9b46931f3049db5ec9b72fe0548b81c2562ea1b
diff --git a/tensorflow/python/keras/layers/recurrent_v2.py b/tensorflow/python/keras/layers/recurrent_v2.py
index a2ed714..9794189 100644
--- a/tensorflow/python/keras/layers/recurrent_v2.py
+++ b/tensorflow/python/keras/layers/recurrent_v2.py
@@ -389,7 +389,7 @@
# TODO(b/162616551): Remove all compat statements after 08/20/2020.
# This follows b/161915509 and is mainly to test the stateless Case op.
- if compat.forward_compatible(2020, 8, 20):
+ if compat.forward_compatible(2020, 8, 27):
# The first two attributes are added to support TFLite use case.
supportive_attributes = {
'time_major': time_major,
@@ -483,7 +483,7 @@
if dropout_mask is not None:
inputs = inputs * dropout_mask[0]
- if compat.forward_compatible(2020, 8, 20):
+ if compat.forward_compatible(2020, 8, 27):
gru_kwargs = {
'inputs': inputs,
'init_h': _read_variable_value(initial_state[0]),
@@ -797,7 +797,7 @@
true_fn=cudnn_gru_fn,
false_fn=standard_gru_fn)
- if compat.forward_compatible(2020, 8, 20):
+ if compat.forward_compatible(2020, 8, 27):
# Chooses the implementation dynamicly based on the running device.
(last_output, outputs, new_h,
runtime) = control_flow_ops.execute_fn_for_device(
@@ -1141,7 +1141,7 @@
else:
logging.warn(_CUDNN_NOT_AVAILABLE_MSG % self.name)
- if compat.forward_compatible(2020, 8, 20):
+ if compat.forward_compatible(2020, 8, 27):
# The first two attributes are added to support TFLite use case.
supportive_attributes = {
'time_major': time_major,
@@ -1202,7 +1202,7 @@
dropout_mask = self.get_dropout_mask_for_cell(inputs, training, count=4)
if dropout_mask is not None:
inputs = inputs * dropout_mask[0]
- if compat.forward_compatible(2020, 8, 20):
+ if compat.forward_compatible(2020, 8, 27):
lstm_kwargs = {
'inputs':
inputs,
@@ -1633,7 +1633,7 @@
true_fn=cudnn_lstm_fn,
false_fn=stardard_lstm_fn)
- if compat.forward_compatible(2020, 8, 20):
+ if compat.forward_compatible(2020, 8, 27):
# Chooses the implementation dynamicly based on the running device.
(last_output, outputs, new_h, new_c,
runtime) = control_flow_ops.execute_fn_for_device(