Searched defs:expandOp (Results 1 – 3 of 3) sorted by relevance
201 if (auto expandOp = dyn_cast<tensor::ExpandShapeOp>(consumer)) { in runOnOperation() local
261 auto expandOp = collapseOp.getSrc().template getDefiningOp<ExpandOpTy>(); in matchAndRewrite() local [all...]
758 bubbleUpPackOpThroughExpandShape(tensor::ExpandShapeOp expandOp,tensor::PackOp packOp,PatternRewriter & rewriter) bubbleUpPackOpThroughExpandShape() argument 895 pushDownUnPackOpThroughExpandShape(tensor::UnPackOp unPackOp,tensor::ExpandShapeOp expandOp,PatternRewriter & rewriter,ControlPropagationFn controlFn) pushDownUnPackOpThroughExpandShape() argument [all...]