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