Searched defs:unPackOp (Results 1 – 6 of 6) sorted by relevance
895 pushDownUnPackOpThroughExpandShape(tensor::UnPackOp unPackOp,tensor::ExpandShapeOp expandOp,PatternRewriter & rewriter,ControlPropagationFn controlFn) pushDownUnPackOpThroughExpandShape() argument 972 matchAndRewrite(tensor::UnPackOp unPackOp,PatternRewriter & rewriter) const matchAndRewrite() argument 1005 auto unPackOp = operand.get().getDefiningOp<tensor::UnPackOp>(); getUnPackedOperand() local [all...]
358 Location loc = unPackOp->getLoc(); in lowerUnPack() argument [all...]
392 auto unPackOp = linalgOp->getOperand(0).getDefiningOp<UnPackOp>(); matchAndRewrite() local [all...]
480 auto unPackOp = cast<UnPackOp>(op); getIterationDomainTileFromOperandTile() local 525 auto unPackOp = cast<UnPackOp>(op); getTiledImplementationFromOperandTile() local [all...]
4119 hasSameInnerOuterAttribute(PackOp packOp,UnPackOp unPackOp) hasSameInnerOuterAttribute() argument 4133 haveSameTiles(PackOp packOp,UnPackOp unPackOp) haveSameTiles() argument 4197 if (auto unPackOp = packOp.getSource().getDefiningOp<UnPackOp>()) { canonicalize() local 4428 canonicalize(UnPackOp unPackOp,PatternRewriter & rewriter) canonicalize() argument [all...]
1638 auto unPackOp = dyn_cast<tensor::UnPackOp>(*packOrUnpackOps.begin()); apply() local [all...]