Home
last modified time | relevance | path

Searched refs:InitTensorOp (Results 1 – 24 of 24) sorted by relevance

/llvm-project-15.0.7/mlir/test/python/dialects/linalg/
H A Dops.py29 return linalg.InitTensorOp([3, 4], f32)
35 return linalg.InitTensorOp([d0, d1], f32)
41 return linalg.InitTensorOp([], f32)
53 op = linalg.InitTensorOp([3, 4], f32)
99 init_result = linalg.InitTensorOp([4, 8], f32)
134 init_result = linalg.InitTensorOp([4, 8], f32)
160 init_result = linalg.InitTensorOp([4, 8], f32)
186 init = linalg.InitTensorOp([4, 8], f32)
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DInitTensorToAllocTensor.cpp21 struct InitTensorLoweringPattern : public OpRewritePattern<InitTensorOp> {
22 using OpRewritePattern<InitTensorOp>::OpRewritePattern;
24 LogicalResult matchAndRewrite(InitTensorOp op, in matchAndRewrite()
H A DFusePadOpWithLinalgProducer.cpp79 auto initTensor = rewriter.create<linalg::InitTensorOp>( in matchAndRewrite()
H A DElementwiseToLinalg.cpp70 res.push_back(b.create<linalg::InitTensorOp>( in getOrCreateOperandsMatchingResultTypes()
H A DSplitReduction.cpp183 initOrAllocTensor = b.create<linalg::InitTensorOp>( in splitReduction()
346 initOrAllocTensor = b.create<linalg::InitTensorOp>( in splitReductionByScaling()
380 newInputs.push_back(b.create<linalg::InitTensorOp>( in splitReductionByScaling()
H A DHoistPadding.cpp434 Value packedTensor = b.create<linalg::InitTensorOp>( in hoistPaddingOnTensors()
547 b.create<InitTensorOp>(loc, ValueRange{}, paddedTensorType.getShape(), in hoistPaddingOnTensors()
H A DDecomposeLinalgOps.cpp188 Value initTensor = rewriter.create<linalg::InitTensorOp>( in createPeeledGenericOp()
H A DTransforms.cpp700 Value initTensor = rewriter.create<InitTensorOp>( in matchAndRewrite()
783 Value init = rewriter.create<InitTensorOp>( in matchAndRewrite()
H A DDropUnitDims.cpp536 linalg::InitTensorOp::getCanonicalizationPatterns(patterns, context); in populateFoldUnitExtentDimsPatterns()
H A DElementwiseOpFusion.cpp1619 auto definingOp = operandVal.getDefiningOp<InitTensorOp>(); in matchAndRewrite()
1630 Value initTensor = rewriter.create<InitTensorOp>( in matchAndRewrite()
H A DTiling.cpp678 InitTensorOp::getCanonicalizationPatterns(patterns, ctx); in populateLinalgTilingCanonicalizationPatterns()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/
H A DLinalgOps.cpp544 auto newInitOp = rewriter.create<InitTensorOp>( in matchAndRewrite()
1189 void InitTensorOp::build(OpBuilder &b, OperationState &result, in build()
1201 LogicalResult InitTensorOp::verify() { in verify()
1215 Type expectedType = InitTensorOp::inferResultType( in verify()
1230 SmallVector<OpFoldResult> InitTensorOp::getMixedSizes() { in getMixedSizes()
1257 struct ReplaceStaticShapeDims : OpRewritePattern<InitTensorOp> {
1258 using OpRewritePattern<InitTensorOp>::OpRewritePattern;
1260 LogicalResult matchAndRewrite(InitTensorOp op, in matchAndRewrite()
1313 rewriter.replaceOpWithNewOp<linalg::InitTensorOp>( in matchAndRewrite()
1338 Value initTensor = rewriter.create<InitTensorOp>( in matchAndRewrite()
[all …]
/llvm-project-15.0.7/mlir/lib/Conversion/TosaToLinalg/
H A DTosaToLinalgNamed.cpp244 Value initTensor = rewriter.create<linalg::InitTensorOp>( in matchAndRewrite()
271 Value biasInitTensor = rewriter.create<linalg::InitTensorOp>( in matchAndRewrite()
429 Value initTensor = rewriter.create<linalg::InitTensorOp>( in matchAndRewrite()
437 Value biasInitTensor = rewriter.create<linalg::InitTensorOp>( in matchAndRewrite()
530 auto initTensor = rewriter.create<linalg::InitTensorOp>( in matchAndRewrite()
600 auto initTensor = rewriter.create<linalg::InitTensorOp>( in matchAndRewrite()
626 .create<linalg::InitTensorOp>(loc, filteredDims, in matchAndRewrite()
735 Value initTensor = rewriter.create<linalg::InitTensorOp>( in matchAndRewrite()
797 Value poolInitTensor = rewriter.create<linalg::InitTensorOp>( in matchAndRewrite()
807 rewriter.create<linalg::InitTensorOp>(loc, kernel, accETy); in matchAndRewrite()
[all …]
H A DTosaToLinalg.cpp608 initTensors.push_back(rewriter.create<linalg::InitTensorOp>( in elementwiseMatchAndRewriteHelper()
811 .create<linalg::InitTensorOp>(loc, dynDims, reduceShape, in reduceMatchAndRewriteHelper()
1130 auto initTensor = rewriter.create<linalg::InitTensorOp>( in matchAndRewrite()
1244 Value initTensor = rewriter.create<linalg::InitTensorOp>( in matchAndRewrite()
1366 auto initTensor = rewriter.create<linalg::InitTensorOp>( in matchAndRewrite()
1678 Value init = rewriter.create<linalg::InitTensorOp>( in matchAndRewrite()
1732 .create<linalg::InitTensorOp>( in matchAndRewrite()
1802 auto initTensor = rewriter.create<linalg::InitTensorOp>( in matchAndRewrite()
1952 .create<linalg::InitTensorOp>(loc, dynDims, resultTy.getShape(), in matchAndRewrite()
1965 .create<linalg::InitTensorOp>( in matchAndRewrite()
[all …]
/llvm-project-15.0.7/mlir/python/mlir/dialects/
H A D_linalg_ops_ext.py23 class InitTensorOp: class
/llvm-project-15.0.7/mlir/test/Dialect/Linalg/
H A Done-shot-bufferize-analysis-init-tensor-elimination.mlir4 // InitTensorOp elimination
/llvm-project-15.0.7/mlir/test/python/dialects/linalg/opdsl/
H A Demit_matmul.py62 init_result = linalg.InitTensorOp([4, 8], f32)
/llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Transforms/
H A DSparseTensorRewriting.cpp48 val.getDefiningOp<InitTensorOp>() || in isEmptyInit()
H A DSparsification.cpp313 return val.getDefiningOp<linalg::InitTensorOp>() || in isMaterializing()
/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/
H A DTensorTilingInterfaceImpl.cpp32 Value initTensor = b.create<linalg::InitTensorOp>( in getDestinationOperands()
/llvm-project-15.0.7/mlir/test/Dialect/Bufferization/Transforms/
H A Done-shot-bufferize-alloc-tensor-elimination.mlir50 // InitTensorOp replaces the alloc_tensor with an inplace extract_slice.
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/IR/
H A DLinalgOps.td108 InitTensorOp::inferResultType(staticShape, elementType),
/llvm-project-15.0.7/mlir/docs/Dialects/Linalg/
H A DOpDSL.md321 tensor_2d = linalg.InitTensorOp([4, 8], f32)
322 tensor_3d = linalg.InitTensorOp([4, 8, 16], f32)
/llvm-project-15.0.7/mlir/test/Integration/Dialect/SparseTensor/taco/tools/
H A Dmlir_pytaco.py902 tensor = linalg.InitTensorOp(self.dst_dims, ir_type).result