Home
last modified time | relevance | path

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

/llvm-project/mlir/unittests/Dialect/SparseTensor/
H A DMergerTest.cpp213 const auto &tensorExp = merger.exp(e); in compareExpression() local
214 if (tensorExp.kind != pattern.kind) in compareExpression()
216 switch (tensorExp.kind) { in compareExpression()
219 return tensorExp.tensor == pattern.tid; in compareExpression()
263 return compareExpression(tensorExp.children.e0, pattern.children.e0); in compareExpression()
288 return compareExpression(tensorExp.children.e0, pattern.children.e0) && in compareExpression()
289 compareExpression(tensorExp.children.e1, pattern.children.e1); in compareExpression()
291 bool eq = compareExpression(tensorExp.children.e0, pattern.children.e0); in compareExpression()
292 if (eq && tensorExp.children.e1 != sparse_tensor::detail::kInvalidId) in compareExpression()
293 return compareExpression(tensorExp.children.e1, pattern.children.e1); in compareExpression()
/llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/Utils/
H A DCodegenEnv.h48 ExprId getExprId() const { return tensorExp; } in getExprId()
207 ExprId tensorExp;
203 ExprId tensorExp; global() variable
H A DCodegenEnv.cpp58 tensorExp = *optExp; in initTensorExp()