Searched refs:_gather_states (Results 1 – 1 of 1) sorted by relevance
563 final_cell_state = self._gather_states(mod_cell_states, sequence_length,565 final_output = self._gather_states(mod_outputs, sequence_length,579 def _gather_states(self, data, indices, batch_size): member in LSTMBlockWrapper