Searched refs:get_step_input_shape (Results 1 – 1 of 1) sorted by relevance
557 def get_step_input_shape(shape): function578 step_input_shape = get_step_input_shape(input_shape)585 step_input_shape = nest.map_structure(get_step_input_shape, input_shape)