Lines Matching defs:expandOp

757 bubbleUpPackOpThroughExpandShape(tensor::ExpandShapeOp expandOp,
770       expandOp.getReassociationIndices();
812       expandOp.getSrcType(), packOp.getStaticInnerTiles(),
821       rewriter, packOp.getLoc(), expandOp.getSrc(), packOp.getMixedTiles(),
824       packOp.getLoc(), expandOp.getSrc(), destTensor, projectedInnerDimsPos,
894     tensor::UnPackOp unPackOp, tensor::ExpandShapeOp expandOp,
897   if (!controlFn(&expandOp.getSrcMutable()))
904   auto expandTy = dyn_cast<RankedTensorType>(expandOp.getType());
909       expandOp.getReassociationIndices();
949       expandOp.getLoc(), newExpandType, unPackOp.getSource(),
958   rewriter.replaceOp(expandOp, newUnPackOp);