Searched refs:_get_dataset_attributes (Results 1 – 2 of 2) sorted by relevance
449 batch_size, _, prefetch_buffer = input_lib._get_dataset_attributes(dataset)
1922 def _get_dataset_attributes(dataset): function