Home | History | Annotate | Download | only in kernel_tests

Lines Matching refs:dataset

28     dataset = dataset_ops.Dataset.range(10).shard(5, 2)
29 iterator = dataset.make_one_shot_iterator()
38 dataset_a = dataset_ops.Dataset.range(10)
39 dataset_b = dataset_ops.Dataset.range(10, 0, -1)
40 dataset = dataset_ops.Dataset.zip((dataset_a, dataset_b)).shard(5, 2)
41 iterator = dataset.make_one_shot_iterator()
50 dataset = dataset_ops.Dataset.range(10).shard(5, 0)
51 iterator = dataset.make_one_shot_iterator()
61 dataset_ops.Dataset.range(10).shard(5, 7)
65 dataset_ops.Dataset.range(10).shard(5, -3)
69 dataset_ops.Dataset.range(10).shard(-3, 1)
73 dataset_ops.Dataset.range(10).shard(0, 1)
76 dataset = dataset_ops.Dataset.range(1).shard(5, 2)
77 iterator = dataset.make_one_shot_iterator()
84 dataset = dataset_ops.Dataset.range(10).shard(7, 5)
85 iterator = dataset.make_one_shot_iterator()
92 dataset = dataset_ops.Dataset.range(10).shard(5, 4)
93 iterator = dataset.make_one_shot_iterator()
101 dataset = dataset_ops.Dataset.range(10).shard(4, 3)
102 iterator = dataset.make_one_shot_iterator()