Searched refs:indices_dataset (Results 1 – 1 of 1) sorted by relevance
282 indices_dataset = dataset_ops.DatasetV2.range(1)284 indices_dataset = indices_dataset.repeat(epochs)299 indices_dataset = indices_dataset.map(permutation).prefetch(1)333 indices_dataset = indices_dataset.flat_map(slice_batch_indices)335 dataset = self.slice_inputs(indices_dataset, inputs)344 def slice_inputs(self, indices_dataset, inputs): argument360 indices_dataset,449 def slice_inputs(self, indices_dataset, inputs): argument491 dataset = indices_dataset.map(