Searched refs:insert_timesteps (Results 1 – 1 of 1) sorted by relevance
203 def insert_timesteps(dims): function207 return nest.map_structure(insert_timesteps, child_output_shape)