internal change

PiperOrigin-RevId: 229136654
This commit is contained in:
A. Unique TensorFlower 2019-01-14 00:47:08 -08:00 committed by TensorFlower Gardener
parent ac4d368293
commit 2a50782fb6

View File

@ -184,7 +184,7 @@ class TFLiteLSTMCell(rnn_cell_impl.LayerRNNCell):
var = self.add_variable(
name, shape=shape, initializer=initializer, partitioner=partitioner)
return self._tflite_wrapper.add_input(
var, name="name", index_override=index)
var, name=name, index_override=index)
weight_initializer = self._initializer
if self.dtype is None: