Searched refs:StorageLayout (Results 1 – 5 of 5) sorted by relevance
114 class StorageLayout {116 explicit StorageLayout(const SparseTensorType &stt) in StorageLayout() function117 : StorageLayout(stt.getEncoding()) {} in StorageLayout()118 explicit StorageLayout(SparseTensorEncodingAttr enc) : enc(enc) { in StorageLayout() function161 return StorageLayout(enc).getNumFields(); in getNumFieldsFromEncoding()165 return StorageLayout(enc).getNumDataFields(); in getNumDataFieldsFromEncoding()173 return StorageLayout(enc).foreachField(callback); in foreachFieldInSparseTensor()
71 `sparse_tensor::StorageLayout`.
147 StorageLayout getLayout() const { return layout; } in getLayout() 152 StorageLayout layout;
95 // SparseTensorDialect StorageLayout.102 void StorageLayout::foreachField( in foreachField() 167 StorageLayout(stt).foreachField([specType, posMemType, crdMemType, valMemType, in foreachFieldAndTypeInSparseTensor() 185 unsigned StorageLayout::getNumFields() const { in getNumFields() 195 unsigned StorageLayout::getNumDataFields() const { in getNumDataFields() 209 StorageLayout::getFieldIndexAndStride(SparseTensorFieldKind kind, in getFieldIndexAndStride() 1319 StorageLayout layout(stt.getEncoding()); in verifyPackUnPack()
263 StorageLayout layout(enc); in matchAndRewrite()