Searched refs:input_dimension_numbers (Results 1 – 1 of 1) sorted by relevance
1358 auto input_dimension_numbers = input_shape.layout().minor_to_major(); in AlignLayouts() local1362 int64 current_dimension_number = input_dimension_numbers[i]; in AlignLayouts()1384 if (dimension_to_alignment_index[input_dimension_numbers[i]] != in AlignLayouts()1386 input_dimension_numbers[i] > current_dimension_number) { in AlignLayouts()1389 current_dimension_number = input_dimension_numbers[i]; in AlignLayouts()