Searched refs:indices_dataset (Results 1 – 1 of 1) sorted by relevance
303 indices_dataset = dataset_ops.DatasetV2.range(1)305 indices_dataset = indices_dataset.repeat(epochs)320 indices_dataset = indices_dataset.map(permutation).prefetch(1)354 indices_dataset = indices_dataset.flat_map(slice_batch_indices)356 dataset = self.slice_inputs(indices_dataset, inputs)365 def slice_inputs(self, indices_dataset, inputs): argument381 indices_dataset,470 def slice_inputs(self, indices_dataset, inputs): argument512 dataset = indices_dataset.map(