Example #1
0
 def __call__(self, inputs, state, scope=None):
   assert not isinstance(inputs, (list, tuple))
   _, _, _, _, new_h = gru_ops.gru_cell(
       cell_size=self._num_units, sequence_len=self._sequence_len,
       h_prev=state, x=inputs, scope=scope)
   return new_h, new_h