================ @@ -253,7 +253,10 @@ class SparseTensorType { CrdTransDirectionKind::dim2lvl); } - /// Returns the Level-shape. + /// Returns the batched level rank. ---------------- aartbik wrote: For consistentcy level-rank https://github.com/llvm/llvm-project/pull/83898