Searched refs:collapseShapeOp (Results 1 – 1 of 1) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/Transforms/ |
| H A D | BufferizableOpInterfaceImpl.cpp | 112 auto collapseShapeOp = cast<tensor::CollapseShapeOp>(op); in bufferize() local
113 RankedTensorType tensorResultType = collapseShapeOp.getResultType(); in bufferize()
115 getBuffer(rewriter, collapseShapeOp.getSrc(), options); in bufferize()
145 rewriter, op, resultType, buffer, collapseShapeOp.getReassociation()); in bufferize()
153 bufferType, collapseShapeOp.getReassociationIndices()); in bufferize()
158 rewriter, op->getLoc(), collapseShapeOp.getSrc(), in bufferize()
159 analysisState.isTensorYielded(collapseShapeOp.getResult()), options); in bufferize()
163 MemRefType::get(collapseShapeOp.getSrcType().getShape(), in bufferize()
164 collapseShapeOp.getSrcType().getElementType(), in bufferize()
172 rewriter, op, buffer, collapseShapeOp.getReassociationIndices()); in bufferize()
|