Lines Matching defs:collapseShapeOp

144     auto collapseShapeOp = cast<tensor::CollapseShapeOp>(op);
146         collapseShapeOp.getSrc(), options, invocationStack);
151         srcBufferType, collapseShapeOp.getReassociationIndices());
155       RankedTensorType tensorResultType = collapseShapeOp.getResultType();
161         srcBufferType, collapseShapeOp.getReassociationIndices());
166     auto collapseShapeOp = cast<tensor::CollapseShapeOp>(op);
167     RankedTensorType tensorResultType = collapseShapeOp.getResultType();
169         getBuffer(rewriter, collapseShapeOp.getSrc(), options);
198           rewriter, op, resultType, buffer, collapseShapeOp.getReassociation());
206         bufferType, collapseShapeOp.getReassociationIndices());
211           rewriter, op->getLoc(), collapseShapeOp.getSrc(), options);
215           MemRefType::get(collapseShapeOp.getSrcType().getShape(),
216                           collapseShapeOp.getSrcType().getElementType(),
224         rewriter, op, buffer, collapseShapeOp.getReassociationIndices());