Lines Matching refs:_map_fn
120 def _map_fn(x):
124 return dataset_ops.Dataset.range(10).map(_map_fn)
159 def _map_fn(x, y, z):
163 dataset_ops.Dataset.from_tensor_slices(components).map(_map_fn)
176 def _map_fn(x):
180 return dataset_ops.Dataset.range(100).map(_map_fn)
263 def _map_fn(x, y, z):
267 _map_fn, num_parallel_calls=3).repeat(self._num_epochs))
274 def _map_fn(x, y, z):
278 _map_fn, num_parallel_calls=3).repeat(self._num_epochs).prefetch(5))
291 def _map_fn(x):
296 _map_fn, num_parallel_calls=2).prefetch(2)
