| /llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/ |
| H A D | SparseTensorConversion.cpp | 52 static Value genLvlSizeCall(OpBuilder &builder, Location loc, Value tensor, in genLvlSizeCall() argument
64 static Value genDimSizeCall(OpBuilder &builder, Location loc, Value tensor, in genDimSizeCall() argument
76 SparseTensorType stt, Value tensor, in createOrFoldLvlCall() argument
103 createOrFoldDimCall(OpBuilder & builder,Location loc,SparseTensorType stt,Value tensor,Dimension dim) createOrFoldDimCall() argument
115 fillDimSizes(OpBuilder & builder,Location loc,SparseTensorType stt,Value tensor,SmallVectorImpl<Value> & out) fillDimSizes() argument
128 getDimSizes(OpBuilder & builder,Location loc,SparseTensorType stt,Value tensor=Value ()) getDimSizes() argument
155 extractBarePtrFromTensor(OpBuilder & builder,Location loc,Value tensor) extractBarePtrFromTensor() argument
392 Value tensor = NewCallParams(rewriter, loc) matchAndRewrite() local
722 Value tensor = adaptor.getTensor(); matchAndRewrite() local
870 auto tensor = rewriter.create<bufferization::ToTensorOp>(loc, retVal[i]); matchAndRewrite() local
[all...] |
| H A D | Sparsification.cpp | 119 static bool findDepIdxSet(Merger &merger, TensorId tensor, Level lvl, in findDepIdxSet() argument
198 getNumNonTrivialIdxExpOnSparseLvls(AffineMap map,Value tensor) getNumNonTrivialIdxExpOnSparseLvls() argument
306 __anon93cfebb60102(OpBuilder &builder, Location loc, Value memref, Value tensor) genBuffers() argument
759 Value tensor = lhs->get(); genExpand() local
1299 Value tensor = lhs->get(); genResult() local
[all...] |
| H A D | SparseTensorRewriting.cpp | 126 Location loc, ShapedType stp, Value tensor) { in sizesForTensor() argument
723 auto tensor = op.getTensor(); matchAndRewrite() local
[all...] |
| H A D | SparseReinterpretMap.cpp | 249 Value tensor = op->getOpOperand(i).get(); in translateMap() local
|
| /llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/Utils/ |
| H A D | SparseTensorDescriptor.h | 232 getTuple(Value tensor) getTuple() argument
248 getDescriptorFromTensorTuple(Value tensor) getDescriptorFromTensorTuple() argument
255 getMutDescriptorFromTensorTuple(Value tensor,SmallVectorImpl<Value> & fields) getMutDescriptorFromTensorTuple() argument
|
| H A D | LoopEmitter.cpp | 66 static Value genSliceOffset(OpBuilder &builder, Location loc, Value tensor, in genSliceOffset() argument
72 static Value genSliceStride(OpBuilder &builder, Location loc, Value tensor, in genSliceStride() argument
204 Value tensor = tensors[t]; makeLevelIterator() local
240 const Value tensor = tryFoldTensors(tensors[t]); initializeLoopEmit() local
305 const Value tensor = tryFoldTensors(tensors[t]); initializeLoopEmit() local
[all...] |
| H A D | CodegenUtils.cpp | 550 sparse_tensor::genToMemref(OpBuilder &builder, Location loc, Value tensor) { in genToMemref() argument
558 Value tensor, Dimension dim) { in genValMemSize() argument
563 createOrFoldSliceOffsetOp(OpBuilder & builder,Location loc,Value tensor,Dimension dim) createOrFoldSliceOffsetOp() argument
573 createOrFoldSliceStrideOp(OpBuilder & builder,Location loc,Value tensor,Dimension dim) createOrFoldSliceStrideOp() argument
583 genReader(OpBuilder & builder,Location loc,SparseTensorType stt,Value tensor,SmallVectorImpl<Value> & dimSizesValues,Value & dimSizesBuffer) genReader() argument
[all...] |
| H A D | CodegenEnv.cpp | 150 const TensorId tensor = makeTensorId(lhs->getOperandNumber()); in isAdmissibleTensorExp() local
|
| /llvm-project/mlir/test/python/dialects/ |
| H A D | python_test.py | 6 import mlir.dialects.tensor as tensor global() namespace
[all...] |
| H A D | tensor.py | 6 import mlir.dialects.tensor as tensor namespace
|
| /llvm-project/mlir/lib/ExecutionEngine/ |
| H A D | SparseTensorRuntime.cpp | 460 index_type sparseDimSize(void *tensor, index_type d) { in sparseDimSize()
464 void endLexInsert(void *tensor) { in endLexInsert()
468 void delSparseTensor(void *tensor) { in delSparseTensor()
|
| /llvm-project/mlir/lib/Dialect/Tensor/IR/ |
| H A D | ValueBoundsOpInterfaceImpl.cpp | 17 namespace tensor { namespace
[all...] |
| H A D | TensorOps.cpp | 1135 if (Attribute tensor = adaptor.getTensor()) fold() local
1168 if (Attribute tensor = adaptor.getTensor()) { fold() local
2476 createCanonicalRankReducingExtractSliceOp(OpBuilder & b,Location loc,Value tensor,RankedTensorType targetType) createCanonicalRankReducingExtractSliceOp() argument
2857 createCanonicalRankReducingInsertSliceOp(OpBuilder & b,Location loc,Value tensor,Value dest) createCanonicalRankReducingInsertSliceOp() argument
[all...] |
| /llvm-project/mlir/lib/Dialect/Tensor/Transforms/ |
| H A D | FoldTensorSubsetOps.cpp | 29 namespace tensor { global() namespace
[all...] |
| H A D | PackAndUnpackPatterns.cpp | 16 namespace tensor { namespace
[all...] |
| H A D | BufferizableOpInterfaceImpl.cpp | 30 namespace tensor { global() namespace
[all...] |
| /llvm-project/mlir/python/mlir/extras/ |
| H A D | types.py | 121 def tensor(*shape, element_type: Type = None, encoding: Optional[str] = None): global() function
|
| /llvm-project/mlir/lib/Conversion/TosaToSCF/ |
| H A D | TosaToSCF.cpp | 86 static Value createTensorDim(OpBuilder &builder, Location loc, Value tensor, in createTensorDim()
|
| /llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | Fusion.cpp | 202 static void getProducerOfTensor(Value tensor, OpResult &opResult) { in getProducerOfTensor()
|
| /llvm-project/mlir/include/mlir/Dialect/SparseTensor/Utils/ |
| H A D | Merger.h | 93 TensorId tensor; member
346 constexpr TensorId tensor(TensorLoopId b) const { return b % numTensors; } in tensor() function
|
| /llvm-project/mlir/lib/Conversion/ShapeToStandard/ |
| H A D | ShapeToStandard.cpp | 201 Value tensor = in matchAndRewrite() local
574 Value tensor = adaptor.getArg(); in matchAndRewrite() local
|
| /llvm-project/mlir/lib/Dialect/Bufferization/IR/ |
| H A D | BufferizableOpInterface.cpp | 149 Value tensor; in allocateTensorForShapedValue() local
630 ensureToMemrefOpIsValid(Value tensor,Type memrefType) ensureToMemrefOpIsValid() argument
[all...] |
| /llvm-project/mlir/unittests/Dialect/SparseTensor/ |
| H A D | MergerTest.cpp | 139 ExprId tensor(unsigned t) const { in tensor() function in __anon93a878f70111::MergerTestBase
|
| /llvm-project/mlir/lib/Conversion/TosaToLinalg/ |
| H A D | TosaToLinalg.cpp | 592 expandRank(PatternRewriter & rewriter,Location loc,Value tensor,int64_t rank) expandRank() argument
648 getTensorDim(PatternRewriter & rewriter,Location loc,IndexPool & indexPool,Value tensor,int64_t index) getTensorDim() argument
654 getOrFoldTensorDim(PatternRewriter & rewriter,Location loc,IndexPool & indexPool,Value tensor,int64_t index) getOrFoldTensorDim() argument
[all...] |
| /llvm-project/mlir/lib/Dialect/SparseTensor/IR/ |
| H A D | SparseTensorDialect.cpp | 1234 lvlIsInBounds(Level lvl,Value tensor) lvlIsInBounds() argument
1925 build(OpBuilder & builder,OperationState & result,Value tensor,ValueRange initArgs,AffineMapAttr order,function_ref<void (OpBuilder &,Location,ValueRange,Value,ValueRange)> bodyBuilder) build() argument
[all...] |