Searched refs:state_template (Results 1 – 1 of 1) sorted by relevance
163 def state_template(self): member in _FunctionalRnnCell251 tf_state = nest.pack_sequence_as(func_cell.state_template, flat_final_state)258 func_cell.state_template, flat_acc_state)270 _SetShapeFromTemplate(tf_state, func_cell.state_template)