internal change
PiperOrigin-RevId: 229136654
This commit is contained in:
parent
ac4d368293
commit
2a50782fb6
@ -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:
|
||||
|
Loading…
Reference in New Issue
Block a user