Searched refs:cell_states (Results 1 – 2 of 2) sorted by relevance
225 def step(cell_inputs, cell_states): argument226 return self.cell.call(cell_inputs, cell_states, **kwargs)358 def step(cell_inputs, cell_states): argument360 h_tm1 = cell_states[0]756 def step(cell_inputs, cell_states): argument758 h_tm1 = cell_states[0] # previous memory state759 c_tm1 = cell_states[1] # previous carry state
544 cell_states, outputs = self._call_cell(560 [array_ops.expand_dims(initial_cell_state, [0]), cell_states], 0)569 final_cell_state = cell_states[-1]