Searched refs:known_last_dims (Results 1 – 1 of 1) sorted by relevance
122 def _get_output_shape(output_rank, known_last_dims): argument123 return [None] * (output_rank - len(known_last_dims)) + list(known_last_dims)