| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| cardinality_test.py | 24 from tensorflow.python.data.ops import dataset_ops
36 lambda: dataset_ops.Dataset.range(5).batch(2, drop_remainder=True), 2),
38 lambda: dataset_ops.Dataset.range(5).batch(2, drop_remainder=False), 3),
40 lambda: dataset_ops.Dataset.range(5).filter(lambda _: True).batch(2),
42 ("Batch4", lambda: dataset_ops.Dataset.range(5).repeat().batch(2),
44 ("Cache1", lambda: dataset_ops.Dataset.range(5).cache(), 5),
45 ("Cache2", lambda: dataset_ops.Dataset.range(5).cache("foo"), 5),
46 ("Concatenate1", lambda: dataset_ops.Dataset.range(5).concatenate(
47 dataset_ops.Dataset.range(5)), 10),
49 lambda: dataset_ops.Dataset.range(5).filter(lambda _: True).concatenate
[all...] |
| prefetch_to_device_test.py | 23 from tensorflow.python.data.ops import dataset_ops
37 host_dataset = dataset_ops.Dataset.range(10)
42 iterator = dataset_ops.make_one_shot_iterator(device_dataset)
45 self.assertTrue(dataset_ops.get_structure(host_dataset).is_compatible_with(
46 dataset_ops.get_structure(device_dataset)))
47 self.assertTrue(dataset_ops.get_structure(host_dataset).is_compatible_with(
48 dataset_ops.get_structure(iterator)))
62 host_dataset = dataset_ops.Dataset.range(10)
68 iterator = dataset_ops.make_one_shot_iterator(device_dataset)
71 self.assertTrue(dataset_ops.get_structure(host_dataset).is_compatible_with
[all...] |
| copy_to_device_test.py | 24 from tensorflow.python.data.ops import dataset_ops
41 host_dataset = dataset_ops.Dataset.range(10)
46 iterator = dataset_ops.make_one_shot_iterator(device_dataset)
49 self.assertTrue(dataset_ops.get_structure(host_dataset).is_compatible_with(
50 dataset_ops.get_structure(device_dataset)))
51 self.assertTrue(dataset_ops.get_structure(host_dataset).is_compatible_with(
52 dataset_ops.get_structure(iterator)))
66 host_dataset = dataset_ops.Dataset.from_tensors([0, 1, 2, 3])
71 iterator = dataset_ops.make_one_shot_iterator(device_dataset)
74 self.assertTrue(dataset_ops.get_structure(host_dataset).is_compatible_with
[all...] |
| directed_interleave_dataset_test.py | 24 from tensorflow.python.data.ops import dataset_ops
35 selector_dataset = dataset_ops.Dataset.range(10).repeat(100)
37 dataset_ops.Dataset.from_tensors(i).repeat(100) for i in range(10)
63 dataset_ops.Dataset.from_tensors(i).repeat(None)
92 probs_ds = dataset_ops.Dataset.from_tensors(probs).repeat()
98 datasets = [dataset_ops.Dataset.from_tensors(w).repeat() for w in words]
100 choice_dataset = dataset_ops.Dataset.from_tensor_slices(choice_array)
112 [dataset_ops.Dataset.range(10),
113 dataset_ops.Dataset.range(20)],
118 [dataset_ops.Dataset.range(10)
[all...] |
| /external/tensorflow/tensorflow/python/keras/engine/ |
| training_utils_test.py | 25 from tensorflow.python.data.ops import dataset_ops
103 ('Batch', lambda: dataset_ops.Dataset.range(5).batch(2), ValueError),
104 ('Cache', lambda: dataset_ops.Dataset.range(5).cache()),
105 ('Concatenate', lambda: dataset_ops.Dataset.range(5).concatenate(
106 dataset_ops.Dataset.range(5))),
107 ('FlatMap', lambda: dataset_ops.Dataset.range(5).flat_map(
108 lambda _: dataset_ops.Dataset.from_tensors(0)), ValueError),
109 ('Filter', lambda: dataset_ops.Dataset.range(5).filter(lambda _: True)),
112 ('FromTensors', lambda: dataset_ops.Dataset.from_tensors(0)),
114 lambda: dataset_ops.Dataset.from_tensor_slices([0, 0, 0]))
[all...] |
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| options_test.py | 25 from tensorflow.python.data.ops import dataset_ops
32 ds = dataset_ops.Dataset.range(0)
33 self.assertEqual(dataset_ops.Options(), ds.options())
36 options = dataset_ops.Options()
37 ds = dataset_ops.Dataset.range(0).with_options(options).cache()
41 options = dataset_ops.Options()
43 ds = dataset_ops.Dataset.range(0).with_options(options).with_options(
48 options1 = dataset_ops.Options()
50 options2 = dataset_ops.Options()
52 ds = dataset_ops.Dataset.range(0).with_options(options1).with_options
[all...] |
| dataset_test.py | 26 from tensorflow.python.data.ops import dataset_ops
45 dataset = dataset_ops.Dataset.range(10)
75 lambda x: dataset_ops.Dataset.range(0),
85 lambda: dataset_ops.Dataset.from_generator(
88 ("FromTensors", lambda: dataset_ops.Dataset.from_tensors([42])),
89 ("FromTensorSlices", lambda: dataset_ops.Dataset.from_tensors([42])),
90 ("Range", lambda: dataset_ops.Dataset.range(10)),
99 dataset_fn = dataset_ops.Dataset.from_sparse_tensor_slices(
109 lambda: dataset_ops.Dataset.range(0)),
112 lambda: dataset_ops.Dataset.range(0))
[all...] |
| from_tensors_test.py | 25 from tensorflow.python.data.ops import dataset_ops
46 dataset = dataset_ops.Dataset.from_tensors(components)
50 nest.flatten(dataset_ops.get_legacy_output_shapes(dataset)))
65 dataset = dataset_ops.Dataset.from_tensors(components)
69 [shape for shape in dataset_ops.get_legacy_output_shapes(dataset)])
84 dataset = dataset_ops.Dataset.from_tensors(components)
88 ], [shape for shape in dataset_ops.get_legacy_output_shapes(dataset)])
98 dataset = dataset_ops.Dataset.from_tensors(components)
101 dataset_ops.get_legacy_output_types(dataset))
103 dataset_ops.get_legacy_output_shapes(dataset)
[all...] |
| shard_test.py | 21 from tensorflow.python.data.ops import dataset_ops
31 dataset = dataset_ops.Dataset.range(10).shard(5, 2)
35 dataset_a = dataset_ops.Dataset.range(10)
36 dataset_b = dataset_ops.Dataset.range(10, 0, -1)
37 dataset = dataset_ops.Dataset.zip((dataset_a, dataset_b)).shard(5, 2)
41 dataset = dataset_ops.Dataset.range(10).shard(5, 0)
46 dataset = dataset_ops.Dataset.range(10).shard(5, 7)
51 dataset = dataset_ops.Dataset.range(10).shard(5, -3)
56 dataset = dataset_ops.Dataset.range(10).shard(-3, 1)
61 dataset = dataset_ops.Dataset.range(10).shard(0, 1
[all...] |
| range_test.py | 21 from tensorflow.python.data.ops import dataset_ops
31 dataset = dataset_ops.Dataset.range(5)
36 dataset = dataset_ops.Dataset.range(start, stop)
41 dataset = dataset_ops.Dataset.range(start, stop, step)
47 dataset = dataset_ops.Dataset.range(start, stop, step)
52 dataset = dataset_ops.Dataset.range(start, stop, step)
57 dataset = dataset_ops.Dataset.range(start, stop)
62 dataset = dataset_ops.Dataset.range(start, stop, step)
67 dataset = dataset_ops.Dataset.range(start, stop, step)
|
| /external/tensorflow/tensorflow/python/data/util/ |
| traverse_test.py | 21 from tensorflow.python.data.ops import dataset_ops
29 class _TestDataset(dataset_ops.UnaryUnchangedStructureDataset):
36 **dataset_ops.flat_structure(self))
38 temp_variant_tensor, **dataset_ops.flat_structure(self))
46 ds = dataset_ops.Dataset.range(10)
52 ds = dataset_ops.Dataset.range(10).map(math_ops.square)
60 ds1 = dataset_ops.Dataset.range(10)
61 ds2 = dataset_ops.Dataset.range(10)
70 ds1 = dataset_ops.Dataset.range(10)
71 ds2 = dataset_ops.Dataset.range(10
[all...] |
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
| assert_next_dataset_test.py | 22 from tensorflow.python.data.ops import dataset_ops
32 dataset = dataset_ops.Dataset.from_tensors(0).apply(
34 options = dataset_ops.Options()
40 dataset = dataset_ops.Dataset.from_tensors(0).apply(
42 options = dataset_ops.Options()
53 dataset = dataset_ops.Dataset.from_tensors(0).apply(
55 options = dataset_ops.Options()
|
| choose_fastest_dataset_test.py | 24 from tensorflow.python.data.ops import dataset_ops
36 dataset = dataset_ops.Dataset.from_tensor_slices([0, 1, 2, 3, 4])
41 expected_shapes=dataset_ops.get_legacy_output_shapes(dataset))
44 dataset = dataset_ops.Dataset.from_tensor_slices([0, 1, 2, 3, 4])
49 expected_shapes=dataset_ops.get_legacy_output_shapes(dataset))
52 dataset = dataset_ops.Dataset.range(600)
62 expected_shapes=dataset_ops.get_legacy_output_shapes(dataset_a))
71 dataset_a = dataset_ops.Dataset.from_tensor_slices(slices_a)
72 dataset_b = dataset_ops.Dataset.from_tensor_slices(slices_b)
|
| optimize_dataset_test.py | 31 from tensorflow.python.data.ops import dataset_ops
51 return dataset_ops.Dataset.from_tensors(0).map(lambda x: x + var)
54 return dataset_ops.Dataset.from_tensors(
55 0).flat_map(lambda _: dataset_ops.Dataset.from_tensors(var))
58 return dataset_ops.Dataset.from_tensors(0).filter(lambda x: x < var)
65 return dataset_ops.Dataset.from_tensors(0).apply(
73 return dataset_ops.Dataset.range(5).apply(
80 return dataset_ops.Dataset.from_tensors(var)
82 return dataset_ops.Dataset.from_tensors(0).repeat(10).apply(
86 return dataset_ops.Dataset.from_tensors(0).apply
[all...] |
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| flat_map_dataset_serialization_test.py | 21 from tensorflow.python.data.ops import dataset_ops
42 return dataset_ops.Dataset.range(x, x + 5)
44 return dataset_ops.Dataset.range(start, start + 5 * 5, 5).flat_map(map_fn)
57 return dataset_ops.Dataset.range(100).map(map_fn)
59 return dataset_ops.Dataset.range(5).flat_map(flat_map_fn)
73 return dataset_ops.Dataset.from_tensor_slices([defun_fn(x)])
75 return dataset_ops.Dataset.range(100).flat_map(map_fn)
84 return dataset_ops.Dataset.range(5).flat_map(
85 lambda _: dataset_ops.Dataset.from_tensor_slices([test_var]))
99 return dataset_ops.Dataset.range(100).map(map_fn
[all...] |
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| optimization.py | 20 from tensorflow.python.data.ops import dataset_ops
63 return dataset_ops._ModelDataset(dataset) # pylint: disable=protected-access
98 return dataset_ops._OptimizeDataset(dataset, optimizations) # pylint: disable=protected-access
103 class _AssertNextDataset(dataset_ops.UnaryUnchangedStructureDataset):
117 **dataset_ops.flat_structure(self)))
121 class _NonSerializableDataset(dataset_ops.UnaryUnchangedStructureDataset):
130 **dataset_ops.flat_structure(self)))
134 class _ChooseFastestDataset(dataset_ops.DatasetV2):
170 **dataset_ops.flat_structure(self)))
181 class _ChooseFastestBranchDataset(dataset_ops.UnaryDataset)
[all...] |
| writers.py | 20 from tensorflow.python.data.ops import dataset_ops
51 if not isinstance(dataset, dataset_ops.DatasetV2):
53 if not dataset_ops.get_structure(dataset).is_compatible_with(
58 dataset_ops.get_legacy_output_shapes(dataset),
59 dataset_ops.get_legacy_output_types(dataset)))
|
| distribute.py | 20 from tensorflow.python.data.ops import dataset_ops
24 class _AutoShardDataset(dataset_ops.UnaryDataset):
51 **dataset_ops.flat_structure(self))
|
| enumerate_ops.py | 22 from tensorflow.python.data.ops import dataset_ops
57 return dataset_ops.Dataset.zip((dataset_ops.Dataset.range(start, max_value),
|
| error_ops.py | 20 from tensorflow.python.data.ops import dataset_ops
55 class _IgnoreErrorsDataset(dataset_ops.UnaryUnchangedStructureDataset):
64 **dataset_ops.flat_structure(self)))
|
| get_single_element.py | 20 from tensorflow.python.data.ops import dataset_ops
61 if not isinstance(dataset, dataset_ops.DatasetV2):
67 dataset._variant_tensor, **dataset_ops.flat_structure(dataset)))
|
| sleep.py | 20 from tensorflow.python.data.ops import dataset_ops
24 class _SleepDataset(dataset_ops.UnaryUnchangedStructureDataset):
33 **dataset_ops.flat_structure(self))
|
| /external/tensorflow/tensorflow/python/data/benchmarks/ |
| range_benchmark.py | 21 from tensorflow.python.data.ops import dataset_ops
30 options = dataset_ops.Options()
32 dataset = dataset_ops.Dataset.range(num_elements)
|
| /external/tensorflow/tensorflow/contrib/data/python/kernel_tests/ |
| assert_element_shape_test.py | 24 from tensorflow.python.data.ops import dataset_ops
43 dataset = dataset_ops.Dataset.range(5).map(create_dataset)
47 dataset_ops.get_legacy_output_shapes(dataset))
51 dataset_ops.get_legacy_output_shapes(result))
53 iterator = dataset_ops.make_initializable_iterator(result)
69 dataset = dataset_ops.Dataset.range(3).map(create_dataset)
85 dataset = dataset_ops.Dataset.range(5).map(create_unknown_shape_dataset)
89 dataset_ops.get_legacy_output_shapes(dataset))
95 dataset_ops.get_legacy_output_shapes(result))
97 iterator = dataset_ops.make_initializable_iterator(result
[all...] |
| /external/tensorflow/tensorflow/python/data/ |
| __init__.py | 26 from tensorflow.python.data.ops.dataset_ops import Dataset
27 from tensorflow.python.data.ops.dataset_ops import make_initializable_iterator
28 from tensorflow.python.data.ops.dataset_ops import make_one_shot_iterator
|
