Lines Matching defs:SparseTensorFieldKind

103     llvm::function_ref<bool(FieldIndex, SparseTensorFieldKind, Level,
116       if (!(callback(fieldIdx++, SparseTensorFieldKind::PosMemRef, l, lt)))
120       if (!(callback(fieldIdx++, SparseTensorFieldKind::CrdMemRef, l, lt)))
140   if (!(callback(fieldIdx++, SparseTensorFieldKind::ValMemRef, kInvalidLevel,
144   if (!(callback(fieldIdx++, SparseTensorFieldKind::StorageSpec, kInvalidLevel,
151     llvm::function_ref<bool(Type, FieldIndex, SparseTensorFieldKind, Level,
169                                              SparseTensorFieldKind fieldKind,
172     case SparseTensorFieldKind::StorageSpec:
174     case SparseTensorFieldKind::PosMemRef:
176     case SparseTensorFieldKind::CrdMemRef:
178     case SparseTensorFieldKind::ValMemRef:
187   foreachField([&numFields](FieldIndex, SparseTensorFieldKind, Level,
197   foreachField([&numFields](FieldIndex fidx, SparseTensorFieldKind, Level,
209 StorageLayout::getFieldIndexAndStride(SparseTensorFieldKind kind,
213   if (kind == SparseTensorFieldKind::CrdMemRef) {
223                                       SparseTensorFieldKind fKind, Level fLvl,
226         (kind == fKind && fKind == SparseTensorFieldKind::ValMemRef)) {
1283 static Type getFieldElemType(SparseTensorType stt, SparseTensorFieldKind kind) {
1285   case SparseTensorFieldKind::CrdMemRef:
1287   case SparseTensorFieldKind::PosMemRef:
1289   case SparseTensorFieldKind::ValMemRef:
1291   case SparseTensorFieldKind::StorageSpec:
1326                        lvlTps](FieldIndex fid, SparseTensorFieldKind fKind,
1328     if (fKind == SparseTensorFieldKind::StorageSpec)
1332     if (fKind == SparseTensorFieldKind::ValMemRef) {