Searched refs:getSparseTensorEncoding (Results 1 – 11 of 11) sorted by relevance
464 auto encDst = getSparseTensorEncoding(dstTp); in genSparse2SparseReshape()465 auto encSrc = getSparseTensorEncoding(srcTp); in genSparse2SparseReshape()565 auto encDst = getSparseTensorEncoding(op.getType()); in matchAndRewrite()585 auto encDst = getSparseTensorEncoding(dstType); in matchAndRewrite()586 auto encSrc = getSparseTensorEncoding(srcType); in matchAndRewrite()603 auto enc = getSparseTensorEncoding(resType); in matchAndRewrite()631 auto enc = getSparseTensorEncoding(resType); in matchAndRewrite()672 auto encDst = getSparseTensorEncoding(resType); in matchAndRewrite()673 auto encSrc = getSparseTensorEncoding(srcType); in matchAndRewrite()990 auto enc = getSparseTensorEncoding(srcType); in matchAndRewrite()[all …]
34 if (auto enc = getSparseTensorEncoding(op->get().getType())) { in isSparseTensor()205 auto encDst = getSparseTensorEncoding(op.getResult().getType()); in matchAndRewrite()206 auto encSrc = getSparseTensorEncoding(op.getSrc().getType()); in matchAndRewrite()
26 if (getSparseTensorEncoding(t)) in containsSparseTensor()
194 auto enc = getSparseTensorEncoding(t->get().getType()); in findSparseAnnotations()272 auto enc = getSparseTensorEncoding(t->get().getType()); in computeIterationGraph()328 auto enc = getSparseTensorEncoding(lhs->get().getType()); in isAdmissableTensorExp()506 auto enc = getSparseTensorEncoding(t->get().getType()); in genBuffers()686 auto enc = getSparseTensorEncoding(t->get().getType()); in genIndex()699 auto enc = getSparseTensorEncoding(t->get().getType()); in genSubscript()1003 auto enc = getSparseTensorEncoding(t->get().getType()); in genInvariants()1172 if (!getSparseTensorEncoding(t->get().getType())) { in denseUnitStrides()1690 if (getSparseTensorEncoding(resType)) { in genResult()1772 auto srcEnc = getSparseTensorEncoding(tval.getType()); in resolveCycle()
77 if (getSparseTensorEncoding(type) != nullptr) in convertSparseTensorTypes()
32 SparseTensorEncodingAttr getSparseTensorEncoding(Type type);
97 : CPred<"!!::mlir::sparse_tensor::getSparseTensorEncoding($_self)">;
184 mlir::sparse_tensor::getSparseTensorEncoding(Type type) { in getSparseTensorEncoding() function in mlir::sparse_tensor237 auto e = getSparseTensorEncoding(getTensor().getType()); in verify()246 auto e = getSparseTensorEncoding(getTensor().getType()); in verify()
262 if (sparse_tensor::getSparseTensorEncoding(getType())) { in verify()
1150 if (sparse_tensor::getSparseTensorEncoding(returnType) || in matchAndRewrite()1151 sparse_tensor::getSparseTensorEncoding(resultType)) in matchAndRewrite()
1615 if (sparse_tensor::getSparseTensorEncoding(operandVal.getType())) in matchAndRewrite()