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