Update on "Reuse types in _named_data_store and support tensor layouts"

Reuse `DataEntry` from data_serializer.py, in _named_data_store.py.

Motivation
- deserialize from flat tensor to named data store output
- support tensor layout in named data store

Differential Revision: [D83490345](https://our.internmc.facebook.com/intern/diff/D83490345/)

[ghstack-poisoned]

Author: lucylq