Lines Matching refs:AllocTensorOp
152 LogicalResult AllocTensorOp::bufferize(RewriterBase &rewriter, in bufferize()
223 bool AllocTensorOp::isMemoryWrite(OpResult opResult, in isMemoryWrite()
229 bool AllocTensorOp::bufferizesToMemoryRead(OpOperand &opOperand, in bufferizesToMemoryRead()
236 bool AllocTensorOp::bufferizesToMemoryWrite(OpOperand &opOperand, in bufferizesToMemoryWrite()
244 AllocTensorOp::getAliasingOpResult(OpOperand &opOperand, in getAliasingOpResult()
250 LogicalResult AllocTensorOp::verify() { in verify()
272 void AllocTensorOp::build(OpBuilder &builder, OperationState &result, in build()
278 void AllocTensorOp::build(OpBuilder &builder, OperationState &result, in build()
297 struct ReplaceStaticShapeDims : OpRewritePattern<AllocTensorOp> {
298 using OpRewritePattern<AllocTensorOp>::OpRewritePattern;
300 LogicalResult matchAndRewrite(AllocTensorOp op, in matchAndRewrite()
322 auto newOp = rewriter.create<AllocTensorOp>( in matchAndRewrite()
335 auto allocTensorOp = dimOp.getSource().getDefiningOp<AllocTensorOp>(); in matchAndRewrite()
347 void AllocTensorOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns()
352 LogicalResult AllocTensorOp::reifyResultShapes( in reifyResultShapes()
365 ParseResult AllocTensorOp::parse(OpAsmParser &parser, OperationState &result) { in parse()
390 result.addAttribute(AllocTensorOp::getOperandSegmentSizeAttr(), in parse()
397 void AllocTensorOp::print(OpAsmPrinter &p) { in print()
402 AllocTensorOp::getOperandSegmentSizeAttr()}); in print()
411 Value AllocTensorOp::getDynamicSize(OpBuilder &b, unsigned idx) { in getDynamicSize()