Lines Matching defs:sparse_tensor

38                                          mlir::sparse_tensor::Level &,
39 mlir::sparse_tensor::Level &);
40 static void printLevelRange(mlir::AsmPrinter &, mlir::sparse_tensor::Level,
41 mlir::sparse_tensor::Level);
47 using namespace mlir::sparse_tensor;
51 namespace mlir::sparse_tensor {
55 } // namespace mlir::sparse_tensor
149 void sparse_tensor::foreachFieldAndTypeInSparseTensor(
950 Level mlir::sparse_tensor::SparseTensorEncodingAttr::getAoSCOOStart() const {
960 mlir::sparse_tensor::SparseTensorEncodingAttr::getCOOSegments() const {
995 bool mlir::sparse_tensor::SparseTensorType::isCOOType(Level startLvl,
1011 mlir::sparse_tensor::SparseTensorType::getCOOType(bool ordered) const {
1036 mlir::sparse_tensor::getSparseTensorEncoding(Type type) {
1044 AffineMap mlir::sparse_tensor::inferLvlToDim(AffineMap dimToLvl,
1059 AffineMap mlir::sparse_tensor::inverseBlockSparsity(AffineMap dimToLvl,
1112 SmallVector<unsigned> mlir::sparse_tensor::getBlockSize(AffineMap dimToLvl) {
1129 bool mlir::sparse_tensor::isBlockSparsity(AffineMap dimToLvl) {
1174 bool mlir::sparse_tensor::hasAnyNonIdentityOperandsOrResults(Operation *op) {
1184 Dimension mlir::sparse_tensor::toDim(SparseTensorEncodingAttr enc, Level l) {
1193 Level mlir::sparse_tensor::toLvl(SparseTensorEncodingAttr enc, Dimension d) {
1756 << " region must end with sparse_tensor.yield";
2251 // parse ": sparse_tensor.iter_space -> ret"
2263 "expected sparse_tensor.iter_space type for "
2319 // parse ": (sparse_tensor.iter_space, ...) -> ret"
2596 return cast<sparse_tensor::YieldOp>(
2710 return cast<sparse_tensor::YieldOp>(