Searched refs:current_dimension_number (Results 1 – 1 of 1) sorted by relevance
1362 int64 current_dimension_number = input_dimension_numbers[i]; in AlignLayouts() local1365 CHECK_NE(input_shape.dimensions(current_dimension_number), 1); in AlignLayouts()1367 dimension_to_alignment_index[current_dimension_number]; in AlignLayouts()1386 input_dimension_numbers[i] > current_dimension_number) { in AlignLayouts()1389 current_dimension_number = input_dimension_numbers[i]; in AlignLayouts()