experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Make caching to be by default True.

Browse Source 
PiperOrigin-RevId: 286649524
Change-Id: I48a2a66fe3983c49cf42f15683b9f3a8ed84188d
This commit is contained in:
Zhenyu Tan 2019-12-20 15:24:09 -08:00 committed by TensorFlower Gardener
parent f4eb7f642d
commit 431e0a8bc7
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/python/keras/layers/recurrent.py
View File

@ -1270,7 +1270,7 @@ class SimpleRNNCell(DropoutRNNCellMixin, Layer):
dropout=0.,
recurrent_dropout=0.,
**kwargs):
self._enable_caching_device = kwargs.pop('enable_caching_device', False)
self._enable_caching_device = kwargs.pop('enable_caching_device', True)
super(SimpleRNNCell, self).__init__(**kwargs)
self.units = units
self.activation = activations.get(activation)
@ -1701,7 +1701,7 @@ class GRUCell(DropoutRNNCellMixin, Layer):
implementation=1,
reset_after=False,
**kwargs):
self._enable_caching_device = kwargs.pop('enable_caching_device', False)
self._enable_caching_device = kwargs.pop('enable_caching_device', True)
super(GRUCell, self).__init__(**kwargs)
self.units = units
self.activation = activations.get(activation)
@ -2255,7 +2255,7 @@ class LSTMCell(DropoutRNNCellMixin, Layer):
recurrent_dropout=0.,
implementation=1,
**kwargs):
self._enable_caching_device = kwargs.pop('enable_caching_device', False)
self._enable_caching_device = kwargs.pop('enable_caching_device', True)
super(LSTMCell, self).__init__(**kwargs)
self.units = units
self.activation = activations.get(activation)