Searched defs:dataset_factory (Results 1 – 1 of 1) sorted by relevance
246 def _testOptimization(self, map_fn, dataset_factory, num_parallel_calls): argument257 dataset_factory = lambda: dataset_ops.Dataset.from_tensors(data).repeat(5) function266 dataset_factory = lambda: dataset_ops.Dataset.from_tensors(data).repeat(5) function276 dataset_factory = lambda: dataset_ops.Dataset.from_tensors(data).repeat(5) function285 dataset_factory = lambda: dataset_ops.Dataset.from_tensors(data).repeat(5) function293 dataset_factory = lambda: dataset_ops.Dataset.from_tensors(1) function307 dataset_factory = lambda: dataset_ops.Dataset.from_tensors(data).repeat(5) function316 dataset_factory = lambda: dataset_ops.Dataset.from_tensors(data).repeat(5) function325 dataset_factory = lambda: dataset_ops.Dataset.from_tensors(data).repeat(5) function336 dataset_factory = lambda: dataset_ops.Dataset.from_tensor_slices(x) function[all …]