Searched refs:tensorExps (Results 1 – 2 of 2) sorted by relevance
/llvm-project/mlir/include/mlir/Dialect/SparseTensor/Utils/ |
H A D | Merger.h | 543 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 D | Merger.cpp | 249 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 …]
|