Home
last modified time | relevance | path

Searched refs:getOutTensorID (Results 1 – 4 of 4) sorted by relevance

/llvm-project/mlir/unittests/Dialect/SparseTensor/
H A DMergerTest.cpp316 EXPECT_TRUE(merger.getOutTensorID() == tid(2)); in MergerTest3T1L()
334 EXPECT_TRUE(merger.getOutTensorID() == tid(3)); in MergerTest4T1L()
358 EXPECT_TRUE(merger.getOutTensorID() == tid(2)); in MergerTest3T1LD()
380 EXPECT_TRUE(merger.getOutTensorID() == tid(3)); in MergerTest4T1LU()
404 EXPECT_TRUE(merger.getOutTensorID() == tid(2)); in MergerTest3T1LSo()
/llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/Utils/
H A DCodegenEnv.h100 loopEmitter.getOutTensorId() == latticeMerger.getOutTensorID() && in makeTensorLevel()
/llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/
H A DSparsification.cpp1009 const TensorId outTid = env.merger().getOutTensorID(); in getAllTidLvlsInLatPoints()
1228 if (tid != env.merger().getOutTensorID() && in endLoopSeq()
1382 Value val = env.emitter().getValBuffer()[env.merger().getOutTensorID()]; in matchAndRewrite()
/llvm-project/mlir/include/mlir/Dialect/SparseTensor/Utils/
H A DMerger.h363 constexpr TensorId getOutTensorID() const { return outTensor; } in getOutTensorID() function