Searched refs:output_shape_rank (Results 1 – 2 of 2) sorted by relevance
393 int64 output_shape_rank = in InsertDegenerateDims() local396 CHECK_LT(dim_to_insert, output_shape_rank); in InsertDegenerateDims()400 output_shape_dim_bounds.reserve(output_shape_rank); in InsertDegenerateDims()403 for (int64 i = 0; i < output_shape_rank; ++i) { in InsertDegenerateDims()
2883 const int64 output_shape_rank = in ValidateGatherDimensionNumbers() local2888 if (offset_dim < 0 || offset_dim >= output_shape_rank) { in ValidateGatherDimensionNumbers()2893 i, offset_dim, output_shape_rank); in ValidateGatherDimensionNumbers()