Searched refs:input_dim_index (Results 1 – 2 of 2) sorted by relevance
397 int input_dim_index = 0; in ApplyNewAndShrinkMasksToLayout() local401 if (input_dim_index < input_rank && in ApplyNewAndShrinkMasksToLayout()402 shrink_axis_mask[input_dim_index] == 1) { in ApplyNewAndShrinkMasksToLayout()403 input_dim_index++; in ApplyNewAndShrinkMasksToLayout()418 input_dim_index++; in ApplyNewAndShrinkMasksToLayout()
1288 for (int64_t input_dim_index : input_dim_indices) { in ReshapeLeavesDimensionsUnmodified() local1292 unmodified_dims[i].first < input_dim_index) { in ReshapeLeavesDimensionsUnmodified()1296 unmodified_dims[i].first != input_dim_index) { in ReshapeLeavesDimensionsUnmodified()