Searched refs:map_structure_up_to (Results 1 – 11 of 11) sorted by relevance
136 output_shapes = nest.map_structure_up_to(
933 out = nest.map_structure_up_to(941 out = nest.map_structure_up_to(949 out = nest.map_structure_up_to(961 nest.map_structure_up_to(968 out = nest.map_structure_up_to(980 nest.map_structure_up_to(
872 def map_structure_up_to(shallow_tree, func, *inputs, **kwargs): function
185 output_shapes = nest.map_structure_up_to(273 output_shapes = nest.map_structure_up_to(
394 output_shapes = nest.map_structure_up_to(2935 self._padding_values = nest.map_structure_up_to(
402 def map_structure_up_to(shallow_tree, func, *inputs): function
418 out = nest.map_structure_up_to(425 out = nest.map_structure_up_to(
116 return nest.map_structure_up_to(
546 output_shapes = nest.map_structure_up_to(
895 final_outputs = nest.map_structure_up_to(
1078 return nest.map_structure_up_to(shallow_structure,