Lines Matching defs:dropDim

212     int64_t dropDim = oldMaskType.getRank() - newMaskType.getRank();
213     return b.create<vector::ExtractOp>(loc, mask, splatZero(dropDim));
296     int64_t dropDim = oldType.getRank() - newType.getRank();
311         write.getLoc(), write.getVector(), splatZero(dropDim));
345   int64_t dropDim = 1;
449                            loc, operands[it.index()], splatZero(dropDim))
462                                                       splatZero(dropDim));
519     int64_t dropDim = vecType.getRank() - newVecType.getRank();
524             op->getLoc(), operand, splatZero(dropDim)));
552     int64_t dropDim = oldType.getRank() - newType.getRank();
558         std::accumulate(dimSizes.begin(), dimSizes.begin() + dropDim + 1,
561     newDimSizes.append(dimSizes.begin() + dropDim + 1, dimSizes.end());