Home
last modified time | relevance | path

Searched refs:addExp (Results 1 – 3 of 3) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Utils/
H A DMerger.cpp882 return addExp(kInvariant, v); in buildTensorExp()
887 return addExp(kInvariant, v); in buildTensorExp()
899 return addExp(kAbsF, e); in buildTensorExp()
901 return addExp(kAbsC, e); in buildTensorExp()
903 return addExp(kCeilF, e); in buildTensorExp()
907 return addExp(kSqrtF, e); in buildTensorExp()
919 return addExp(kSinF, e); in buildTensorExp()
921 return addExp(kSinC, e); in buildTensorExp()
929 return addExp(kNegC, e); in buildTensorExp()
951 return addExp(kCIm, e); in buildTensorExp()
[all …]
/llvm-project-15.0.7/mlir/unittests/Dialect/SparseTensor/
H A DMergerTest.cpp137 return merger.addExp(Kind::kTensor, tensor); in tensor()
310 merger.addExp(Kind::kTensor, t0, -1u); in MergerTest3T1L()
314 merger.addExp(Kind::kTensor, t1, -1u); in MergerTest3T1L()
318 merger.addExp(Kind::kTensor, t2, -1u); in MergerTest3T1L()
333 merger.addExp(Kind::kTensor, t0, -1u); in MergerTest4T1L()
337 merger.addExp(Kind::kTensor, t1, -1u); in MergerTest4T1L()
341 merger.addExp(Kind::kTensor, t2, -1u); in MergerTest4T1L()
345 merger.addExp(Kind::kTensor, t3, -1u); in MergerTest4T1L()
364 merger.addExp(Kind::kTensor, t0, -1u); in MergerTest3T1LD()
368 merger.addExp(Kind::kTensor, t1, -1u); in MergerTest3T1LD()
[all …]
/llvm-project-15.0.7/mlir/include/mlir/Dialect/SparseTensor/Utils/
H A DMerger.h160 unsigned addExp(Kind k, unsigned e0, unsigned e1 = -1u, Value v = Value(),
162 unsigned addExp(Kind k, unsigned e, Value v, Operation *op = nullptr) {
163 return addExp(k, e, -1u, v, op);
165 unsigned addExp(Kind k, Value v, Operation *op = nullptr) {
166 return addExp(k, -1u, -1u, v, op);