Searched refs:inserted_dim (Results 1 – 2 of 2) sorted by relevance
3481 for (int64 inserted_dim : dim_numbers.inserted_window_dims()) { in ValidateScatterDimensionNumbers() local3482 if (inserted_dim < 0 || inserted_dim >= operand_shape.dimensions_size()) { in ValidateScatterDimensionNumbers()3486 operand_shape.dimensions_size(), inserted_dim); in ValidateScatterDimensionNumbers()
5549 auto inserted_dim = expand_dims[0].getSExtValue(); in matchAndRewrite() local5552 if (inserted_dim < 0) { in matchAndRewrite()5553 inserted_dim += result_ty.getRank(); in matchAndRewrite()5555 if (inserted_dim < 0) { in matchAndRewrite()5560 dims[inserted_dim] = rewriter.create<ConstantIndexOp>(op.getLoc(), 1); in matchAndRewrite()5568 dims[i >= inserted_dim ? i + 1 : i] = dim; in matchAndRewrite()