Home
last modified time | relevance | path

Searched refs:makeTensorId (Results 1 – 6 of 6) sorted by relevance

/llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/Utils/
H A DCodegenEnv.h72 constexpr TensorId makeTensorId(unsigned t) const { in makeLoopId()
73 return latticeMerger.makeTensorId(t); in makeLoopId()
68 constexpr TensorId makeTensorId(unsigned t) const { makeTensorId() function
H A DCodegenEnv.cpp80 const TensorId tid = makeTensorId(t.getOperandNumber()); in startEmit()
150 const TensorId tensor = makeTensorId(lhs->getOperandNumber()); in isAdmissibleTensorExp()
/llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/
H A DSparsification.cpp251 const TensorId tid = env.makeTensorId(t.getOperandNumber()); in findSparseAnnotations()
352 const TensorId tid = env.makeTensorId(t->getOperandNumber()); in genSubscript()
602 const TensorId tid = env.makeTensorId(arg.getArgNumber()); in relinkBranch()
1160 const TensorId tid = env.makeTensorId(input->getOperandNumber()); in translateBitsToTidLvlPairs()
/llvm-project/mlir/include/mlir/Dialect/SparseTensor/Utils/
H A DMerger.h243 constexpr TensorId makeTensorId(unsigned t) const { in makeTensorId() function
/llvm-project/mlir/unittests/Dialect/SparseTensor/
H A DMergerTest.cpp137 TensorId tid(unsigned t) const { return merger.makeTensorId(t); } in tid()
/llvm-project/mlir/lib/Dialect/SparseTensor/Utils/
H A DMerger.cpp1292 const TensorId tid = makeTensorId(arg.getArgNumber()); in buildTensorExp()