Home
last modified time | relevance | path

Searched refs:tensorExps (Results 1 – 2 of 2) sorted by relevance

/llvm-project/mlir/include/mlir/Dialect/SparseTensor/Utils/
H A DMerger.h543 return tensorExps[e]; in exp()
562 tensorExps[e].val = v; in setExprValue()
569 tensorExps[e].val = Value(); in clearExprValue()
604 return e != detail::kInvalidId && e < tensorExps.size(); in isValidExprId()
663 llvm::SmallVector<TensorExp> tensorExps; variable
/llvm-project/mlir/lib/Dialect/SparseTensor/Utils/
H A DMerger.cpp249 const ExprId eNew(tensorExps.size()); in addTensorExp()
250 tensorExps.emplace_back(TensorExp::Kind::kTensor, t, detail::kInvalidId, in addTensorExp()
257 const ExprId eNew(tensorExps.size()); in addLoopVarExp()
258 tensorExps.emplace_back(TensorExp::Kind::kLoopVar, i, detail::kInvalidId, in addLoopVarExp()
264 const ExprId eNew(tensorExps.size()); in addInvariantExp()
265 tensorExps.emplace_back(TensorExp::Kind::kInvariant, detail::kInvalidId, in addInvariantExp()
271 const ExprId eNew(tensorExps.size()); in addSynZeroExp()
272 tensorExps.emplace_back(TensorExp::Kind::kSynZero, detail::kInvalidId, in addSynZeroExp()
280 const ExprId eNew(tensorExps.size()); in addExp()
281 tensorExps.emplace_back(k, e0, e1, Value(), op, attr); in addExp()
[all …]