Searched refs:collapsed_dim (Results 1 – 1 of 1) sorted by relevance
2931 for (int64 collapsed_dim : dim_numbers.collapsed_slice_dims()) { in ValidateGatherDimensionNumbers() local2932 if (collapsed_dim < 0 || collapsed_dim >= input_shape.dimensions_size()) { in ValidateGatherDimensionNumbers()2936 input_shape.dimensions_size(), collapsed_dim); in ValidateGatherDimensionNumbers()