Lines Matching refs:shape
20 using namespace mlir::shape;
23 namespace shape { namespace
29 shape::AssumingOp> {
37 auto assumingOp = cast<shape::AssumingOp>(op); in getAliasingOpOperand()
43 auto yieldOp = dyn_cast<shape::AssumingYieldOp>( in getAliasingOpOperand()
63 auto assumingOp = cast<shape::AssumingOp>(op); in bufferize()
66 auto yieldOp = cast<shape::AssumingYieldOp>( in bufferize()
71 auto newOp = rewriter.create<shape::AssumingOp>( in bufferize()
103 shape::AssumingYieldOp> {
116 assert(isa<shape::AssumingOp>(op->getParentOp()) && in getAliasingOpResult()
131 auto yieldOp = cast<shape::AssumingYieldOp>(op); in bufferize()
143 replaceOpWithNewBufferizedOp<shape::AssumingYieldOp>(rewriter, op, in bufferize()
153 void mlir::shape::registerBufferizableOpInterfaceExternalModels( in registerBufferizableOpInterfaceExternalModels()
155 registry.addExtension(+[](MLIRContext *ctx, shape::ShapeDialect *dialect) { in registerBufferizableOpInterfaceExternalModels()
156 shape::AssumingOp::attachInterface<AssumingOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()
157 shape::AssumingYieldOp::attachInterface<AssumingYieldOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()