keras/src/trainers/data_adapters/tf_dataset_adapter.py/TFDatasetAdapter/get_numpy_iterator
class TFDatasetAdapter:    def get_numpy_iterator(self):
        from keras.src.backend.tensorflow.core import convert_to_numpy

        for batch in self._dataset:
            yield tree.map_structure(convert_to_numpy, batch)