Home | History | Annotate | Download | only in kernel_tests

Lines Matching refs:dataset

69     # Cycle length 1 acts like `Dataset.flat_map()`.
107 dataset = (
108 dataset_ops.Dataset.from_tensor_slices(input_values)
110 .interleave(lambda x: dataset_ops.Dataset.from_tensors(x).repeat(x),
112 iterator = dataset.make_initializable_iterator()
117 # Cycle length 1 acts like `Dataset.flat_map()`.
186 return dataset_ops.Dataset.from_tensor_slices(
190 dataset_ops.Dataset.range(10).map(_map_fn).interleave(
206 dataset_ops.Dataset.from_tensor_slices([])
208 .interleave(dataset_ops.Dataset.from_tensors, cycle_length=2)