Lines Matching refs:dataset
50 dataset = dataset_ops.Dataset.from_tensors(test_case['tensor'])
51 iterator = dataset.make_one_shot_iterator()
75 dataset = dataset_ops.Dataset.from_tensor_slices(test_case['tensor'])
76 iterator = dataset.make_one_shot_iterator()
108 dataset = dataset_ops.Dataset.from_generator(
112 iterator = dataset.make_one_shot_iterator()
124 dataset = dataset_ops.Dataset.range(42)
125 iterator = dataset.make_one_shot_iterator()
149 dataset = dataset_ops.Dataset.from_tensors(test_case['tensor'])
150 dataset = fn(dataset, test_case['tensor'], test_case['shape'])
151 iterator = dataset.make_one_shot_iterator()
163 def fn(dataset, tensor, shape):
165 return dataset.concatenate(dataset_ops.Dataset.from_tensors(tensor))
171 def fn(dataset, tensor, shape):
173 return dataset.prefetch(42)
179 def fn(dataset, tensor, shape):
181 return dataset.repeat(42)
187 def fn(dataset, tensor, shape):
189 return dataset.shuffle(42)
195 def fn(dataset, tensor, shape):
197 return dataset.cache()
203 def fn(dataset, tensor, shape):
205 return dataset.take(42)
211 def fn(dataset, tensor, shape):
213 return dataset.skip(42)
219 def fn(dataset, tensor, shape):
221 return dataset.shard(42, 0)
227 def fn(dataset, tensor, shape):
229 return dataset.filter(lambda x: True)
253 dataset = dataset_ops.Dataset.from_tensors(test_case['tensor'])
254 dataset = dataset.batch(42)
255 iterator = dataset.make_one_shot_iterator()
282 dataset = dataset_ops.Dataset.from_tensors(test_case['tensor'])
283 dataset = dataset.padded_batch(42, padded_shapes=test_case['shape'][1:])
284 iterator = dataset.make_one_shot_iterator()
311 dataset = dataset_ops.Dataset.range(42)
316 return dataset_ops.Dataset.from_tensors(tensor).repeat(n)
320 dataset = dataset.flat_map(make_dataset(test_case['tensor']))
321 iterator = dataset.make_one_shot_iterator()
345 dataset = dataset_ops.Dataset.range(42)
350 return dataset_ops.Dataset.from_tensors(tensor).repeat(n)
354 dataset = dataset.interleave(
356 iterator = dataset.make_one_shot_iterator()
383 dataset = dataset_ops.Dataset.from_tensors(test_case['tensor'])
384 dataset = dataset.map(array_ops.transpose)
385 iterator = dataset.make_one_shot_iterator()
