Searched refs:inserted_dim (Results 1 – 2 of 2) sorted by relevance
3600 for (int64_t inserted_dim : dim_numbers.inserted_window_dims()) { in ValidateScatterDimensionNumbers() local3601 if (inserted_dim < 0 || inserted_dim >= operand_shape.dimensions_size()) { in ValidateScatterDimensionNumbers()3605 operand_shape.dimensions_size(), inserted_dim); in ValidateScatterDimensionNumbers()
6425 auto inserted_dim = expand_dims[0].getSExtValue(); in matchAndRewrite() local6428 if (inserted_dim < 0) { in matchAndRewrite()6429 inserted_dim += result_ty.getRank(); in matchAndRewrite()6431 if (inserted_dim < 0) { in matchAndRewrite()6436 dims[inserted_dim] = in matchAndRewrite()6443 dims[i >= inserted_dim ? i + 1 : i] = dim; in matchAndRewrite()