| /llvm-project-15.0.7/mlir/test/python/dialects/ |
| H A D | tensor.py | 6 import mlir.dialects.tensor as tensor namespace
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Transforms/ |
| H A D | Sparsification.cpp | 475 Value tensor = lhs->get(); in genOutputBuffer() local
697 unsigned tensor = t->getOperandNumber(); in genSubscript() local
1053 Value tensor = lhs->get(); in genExpansion() local
1093 unsigned tensor = merger.tensor(b); in genInit() local
1195 unsigned tensor = merger.tensor(fb); in genFor() local
1273 unsigned tensor = merger.tensor(b); in genWhile() local
1306 unsigned tensor = merger.tensor(b); in genWhile() local
1354 unsigned tensor = merger.tensor(b); in genLocals() local
1384 unsigned tensor = merger.tensor(b); in genLocals() local
1442 unsigned tensor = merger.tensor(b); in genWhileInduction() local
[all …]
|
| H A D | SparseTensorConversion.cpp | 245 Value tensor, Value ind, ValueRange ivs) { in genIndexAndValueForDense()
296 genSplitSparseConstant(OpBuilder &builder, Location loc, Value tensor) { in genSplitSparseConstant()
353 Value elemPtr, Value tensor, in insertScalarIntoDenseTensor()
|
| /llvm-project-15.0.7/mlir/unittests/Dialect/SparseTensor/ |
| H A D | MergerTest.cpp | 136 unsigned tensor(unsigned tensor) { in tensor()
196 auto tensor = std::get<1>(l); in loopsToBits() local
|
| /llvm-project-15.0.7/mlir/test/Integration/Dialect/SparseTensor/taco/tools/ |
| H A D | mlir_pytaco_api.py | 33 tensor = mlir_pytaco.Tensor variable
|
| /llvm-project-15.0.7/mlir/lib/ExecutionEngine/ |
| H A D | SparseTensorUtils.cpp | 587 auto *tensor = in newSparseTensor() local
792 SparseTensorEnumeratorBase(const SparseTensorStorageBase &tensor, in SparseTensorEnumeratorBase()
847 SparseTensorEnumerator(const SparseTensorStorage<P, I, V> &tensor, in SparseTensorEnumerator()
1013 const DimLevelType *sparsity, const SparseTensorStorageBase &tensor) in SparseTensorStorage()
1397 static void outSparseTensor(void *tensor, void *dest, bool sort) { in outSparseTensor()
1456 auto *tensor = SparseTensorStorage<uint64_t, uint64_t, V>::newSparseTensor( in toMLIRSparseTensor() local
1464 static void fromMLIRSparseTensor(void *tensor, uint64_t *pRank, uint64_t *pNse, in fromMLIRSparseTensor()
1807 void endInsert(void *tensor) { in endInsert()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/SparseTensor/Utils/ |
| H A D | Merger.h | 103 unsigned tensor; member
223 unsigned tensor(unsigned b) const { return b % numTensors; } in tensor() function
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/IR/ |
| H A D | SparseTensorDialect.cpp | 194 static LogicalResult isInBounds(Value dim, Value tensor) { in isInBounds()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/IR/ |
| H A D | BufferizableOpInterface.cpp | 70 Value tensor; in allocateTensorForShapedValue() local
539 static void ensureToMemrefOpIsValid(Value tensor, Type memrefType) { in ensureToMemrefOpIsValid()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/ShapeToStandard/ |
| H A D | ShapeToStandard.cpp | 198 Value tensor = in matchAndRewrite() local
551 Value tensor = adaptor.getArg(); in matchAndRewrite() local
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | Fusion.cpp | 379 static void getProducerOfTensor(Value tensor, OpResult &opResult) { in getProducerOfTensor()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SCF/Transforms/ |
| H A D | BufferizableOpInterfaceImpl.cpp | 31 static void ensureToMemrefOpIsValid(Value tensor, Type memrefType) { in ensureToMemrefOpIsValid()
305 static FailureOr<Value> getYieldedBuffer(RewriterBase &rewriter, Value tensor, in getYieldedBuffer()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorOps.cpp | 417 if (Attribute tensor = operands.front()) in fold() local
451 if (Attribute tensor = operands.front()) { in fold() local
1433 OpBuilder &b, Location loc, Value tensor, RankedTensorType targetType) { in createCanonicalRankReducingExtractSliceOp()
1785 Value tensor, in createCanonicalRankReducingInsertSliceOp()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/Transforms/ |
| H A D | BufferizableOpInterfaceImpl.cpp | 24 namespace tensor { namespace
|
| /llvm-project-15.0.7/mlir/lib/IR/ |
| H A D | Operation.cpp | 716 if (auto tensor = type.dyn_cast<TensorType>()) in getTensorOrVectorElementType() local
|