Searched refs:ds_iter (Results 1 – 1 of 1) sorted by relevance
151 ds_iter = iter(adapter.get_dataset())154 next(ds_iter)156 next(ds_iter)282 def _get_epoch(ds_iter): argument285 ds_data.append(next(ds_iter).numpy())288 ds_iter = iter(adapter.get_dataset())291 epoch_data = _get_epoch(ds_iter)298 second_epoch_data = _get_epoch(ds_iter)315 def _get_epoch_batches(ds_iter): argument318 ds_data.append(next(ds_iter)[0].numpy())[all …]