Searched refs:MaybeRemoveOutputFromState (Results 1 – 1 of 1) sorted by relevance
130 state0_flat = self.MaybeRemoveOutputFromState(extended_state0_flat)173 def MaybeRemoveOutputFromState(self, flat_state): member in _FunctionalRnnCell249 flat_final_state = func_cell.MaybeRemoveOutputFromState(255 flat_acc_state = func_cell.MaybeRemoveOutputFromState(