Searched refs:AllocTensorOp (Results 1 – 14 of 14) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/IR/ |
| H A D | BufferizationOps.cpp | 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()
298 using OpRewritePattern<AllocTensorOp>::OpRewritePattern;
300 LogicalResult matchAndRewrite(AllocTensorOp op, in matchAndRewrite()
322 auto newOp = rewriter.create<AllocTensorOp>( in matchAndRewrite()
352 LogicalResult AllocTensorOp::reifyResultShapes( in reifyResultShapes()
[all …]
|
| H A D | BufferizableOpInterface.cpp | 106 auto allocTensorOp = b.create<AllocTensorOp>(loc, tensorType, dynamicSizes, in allocateTensorForShapedValue()
497 if (!tensor.getDefiningOp<AllocTensorOp>()) in isTensorYielded()
|
| /llvm-project-15.0.7/mlir/python/mlir/dialects/ |
| H A D | _bufferization_ops_ext.py | 15 class AllocTensorOp: class
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Transforms/ |
| H A D | SparseTensorPasses.cpp | 128 target.addDynamicallyLegalOp<bufferization::AllocTensorOp>( in runOnOperation()
129 [&](bufferization::AllocTensorOp op) { in runOnOperation()
|
| H A D | SparseTensorRewriting.cpp | 49 val.getDefiningOp<AllocTensorOp>(); in isEmptyInit()
|
| H A D | SparseTensorConversion.cpp | 621 : public OpConversionPattern<bufferization::AllocTensorOp> {
625 matchAndRewrite(bufferization::AllocTensorOp op, OpAdaptor adaptor, in matchAndRewrite()
|
| H A D | Sparsification.cpp | 314 val.getDefiningOp<bufferization::AllocTensorOp>(); in isMaterializing()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | InitTensorToAllocTensor.cpp | 26 rewriter.replaceOpWithNewOp<bufferization::AllocTensorOp>(op, op.getType(), in matchAndRewrite()
|
| H A D | SplitReduction.cpp | 177 initOrAllocTensor = b.create<bufferization::AllocTensorOp>( in splitReduction()
344 b.create<bufferization::AllocTensorOp>(loc, newT, dims); in splitReductionByScaling()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/Transforms/ |
| H A D | AllocTensorElimination.cpp | 139 !maybeAllocTensor.front().getDefiningOp<AllocTensorOp>()) in eliminateAllocTensors()
|
| /llvm-project-15.0.7/mlir/test/Dialect/Bufferization/Transforms/ |
| H A D | one-shot-bufferize-alloc-tensor-elimination.mlir | 20 // insert_slice. AllocTensorOp replaces the alloc_tensor with an out-of-place
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Bufferization/IR/ |
| H A D | BufferizationOps.td | 23 // AllocTensorOp
|
| /llvm-project-15.0.7/mlir/test/Dialect/Linalg/ |
| H A D | one-shot-bufferize.mlir | 342 // AllocTensorOp elimination would produce SSA violations for the example below.
|
| /llvm-project-15.0.7/mlir/test/Integration/Dialect/SparseTensor/taco/tools/ |
| H A D | mlir_pytaco.py | 910 return bufferization.AllocTensorOp(mlir_type, [], None, None)
|