Home | History | Annotate | Download | only in kernel_tests

Lines Matching refs:dataset

37     return dataset_ops.Dataset.from_tensors(0).repeat(None).apply(
61 iterator = dataset_ops.Dataset.from_tensors(1).repeat(None).apply(
85 dataset = dataset_ops.Dataset.from_tensors(0).repeat(5).apply(
87 self.assertEqual([None], dataset.output_shapes[0][0].as_list())
88 self.assertIs(None, dataset.output_shapes[0][1].ndims)
89 self.assertEqual([], dataset.output_shapes[1].as_list())
91 iterator = dataset.make_one_shot_iterator()
107 dataset = dataset_ops.Dataset.range(10)
111 dataset.apply(
119 dataset = dataset_ops.Dataset.range(10)
124 dataset.apply(
132 return dataset_ops.Dataset.from_tensors(1).repeat(num_elements).apply(