Support maps of tensor infos for generating catalog documentation
Support maps of tensor infos for generating catalog documentation

PiperOrigin-RevId: 433503970

tensorflow/datasets,tensorflow/datasets,tensorflow/datasets,tensorflow/datasets,tensorflow/datasets