| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | csv_dataset_test.py | 26 from tensorflow.python.data.kernel_tests import test_base
37 class CsvDatasetTest(test_base.DatasetTestBase, parameterized.TestCase):
100 @combinations.generate(test_base.default_test_combinations())
106 @combinations.generate(test_base.default_test_combinations())
112 @combinations.generate(test_base.default_test_combinations())
118 @combinations.generate(test_base.default_test_combinations())
124 @combinations.generate(test_base.default_test_combinations())
132 @combinations.generate(test_base.default_test_combinations())
141 @combinations.generate(test_base.default_test_combinations())
151 @combinations.generate(test_base.default_test_combinations())
[all …]
|
| D | rebatch_dataset_test.py | 22 from tensorflow.python.data.kernel_tests import test_base
35 class BatchSizesForWorkerTest(test_base.DatasetTestBase,
77 test_base.default_test_combinations(),
96 test_base.default_test_combinations(),
118 test_base.default_test_combinations(),
129 test_base.default_test_combinations(),
140 test_base.default_test_combinations(),
157 class RebatchDatasetTest(test_base.DatasetTestBase, parameterized.TestCase):
163 @combinations.generate(test_base.default_test_combinations())
172 combinations.times(test_base.default_test_combinations(),
[all …]
|
| D | from_list_test.py | 23 from tensorflow.python.data.kernel_tests import test_base
32 class FromListTest(test_base.DatasetTestBase, parameterized.TestCase):
36 test_base.default_test_combinations(),
50 @combinations.generate(test_base.default_test_combinations())
61 test_base.default_test_combinations(),
75 @combinations.generate(test_base.default_test_combinations())
84 @combinations.generate(test_base.default_test_combinations())
103 @combinations.generate(test_base.default_test_combinations())
116 class FromListRandomAccessTest(test_base.DatasetTestBase,
119 @combinations.generate(test_base.default_test_combinations())
[all …]
|
| D | sql_dataset_test.py | 27 from tensorflow.python.data.kernel_tests import test_base
35 class SqlDatasetTestBase(test_base.DatasetTestBase):
105 @combinations.generate(test_base.default_test_combinations())
120 @combinations.generate(test_base.default_test_combinations())
137 @combinations.generate(test_base.default_test_combinations())
154 @combinations.generate(test_base.default_test_combinations())
179 @combinations.generate(test_base.default_test_combinations())
190 @combinations.generate(test_base.default_test_combinations())
201 @combinations.generate(test_base.default_test_combinations())
212 @combinations.generate(test_base.default_test_combinations())
[all …]
|
| D | shuffle_and_repeat_test.py | 21 from tensorflow.python.data.kernel_tests import test_base
29 class ShuffleAndRepeatTest(test_base.DatasetTestBase, parameterized.TestCase):
45 @combinations.generate(test_base.default_test_combinations())
54 @combinations.generate(test_base.default_test_combinations())
63 @combinations.generate(test_base.default_test_combinations())
69 @combinations.generate(test_base.default_test_combinations())
76 @combinations.generate(test_base.default_test_combinations())
85 @combinations.generate(test_base.default_test_combinations())
94 @combinations.generate(test_base.default_test_combinations())
102 @combinations.generate(test_base.default_test_combinations())
[all …]
|
| D | compression_ops_test.py | 20 from tensorflow.python.data.kernel_tests import test_base
80 class CompressionOpsTest(test_base.DatasetTestBase, parameterized.TestCase):
83 combinations.times(test_base.default_test_combinations(),
86 test_base.v2_eager_only_combinations(),
97 combinations.times(test_base.default_test_combinations(),
100 test_base.v2_eager_only_combinations(),
113 combinations.times(test_base.default_test_combinations()))
124 combinations.times(test_base.default_test_combinations()))
138 combinations.times(test_base.default_test_combinations()))
147 combinations.times(test_base.default_test_combinations()))
[all …]
|
| D | io_test.py | 24 from tensorflow.python.data.kernel_tests import test_base
32 class IOTest(test_base.DatasetTestBase, parameterized.TestCase):
53 combinations.times(test_base.eager_only_combinations(),
62 @combinations.generate(test_base.eager_only_combinations())
69 @combinations.generate(test_base.eager_only_combinations())
79 @combinations.generate(test_base.eager_only_combinations())
93 combinations.times(test_base.eager_only_combinations(),
108 @combinations.generate(test_base.eager_only_combinations())
120 @combinations.generate(test_base.graph_only_combinations())
127 @combinations.generate(test_base.eager_only_combinations())
[all …]
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/service/ |
| D | BUILD | 17 ":test_base",
21 "//tensorflow/python/data/kernel_tests:test_base",
38 ":test_base",
45 "//tensorflow/python/data/kernel_tests:test_base",
55 ":test_base",
63 "//tensorflow/python/data/kernel_tests:test_base",
73 ":test_base",
94 "//tensorflow/python/data/kernel_tests:test_base",
105 ":test_base",
117 "//tensorflow/python/data/kernel_tests:test_base",
[all …]
|
| D | auto_shard_test.py | 21 from tensorflow.python.data.experimental.kernel_tests.service import test_base as data_service_test…
25 from tensorflow.python.data.kernel_tests import test_base
68 test_base.default_test_combinations(),
80 @combinations.generate(test_base.default_test_combinations())
94 test_base.default_test_combinations(),
109 @combinations.generate(test_base.default_test_combinations())
125 @combinations.generate(test_base.default_test_combinations())
139 test_base.default_test_combinations(),
156 @combinations.generate(test_base.default_test_combinations())
167 @combinations.generate(test_base.default_test_combinations())
[all …]
|
| D | dynamic_sharding_test.py | 20 from tensorflow.python.data.experimental.kernel_tests.service import test_base as data_service_test…
22 from tensorflow.python.data.kernel_tests import test_base
43 @combinations.generate(test_base.default_test_combinations())
52 @combinations.generate(test_base.default_test_combinations())
60 @combinations.generate(test_base.default_test_combinations())
69 @combinations.generate(test_base.default_test_combinations())
80 @combinations.generate(test_base.default_test_combinations())
89 @combinations.generate(test_base.default_test_combinations())
106 @combinations.generate(test_base.default_test_combinations())
116 @combinations.generate(test_base.default_test_combinations())
[all …]
|
| D | local_workers_test.py | 20 from tensorflow.python.data.experimental.kernel_tests.service import test_base as data_service_test…
22 from tensorflow.python.data.kernel_tests import test_base
31 @combinations.generate(test_base.default_test_combinations())
42 test_base.default_test_combinations(),
59 test_base.default_test_combinations(),
79 test_base.default_test_combinations(),
97 test_base.default_test_combinations(),
115 test_base.default_test_combinations(),
131 @combinations.generate(test_base.default_test_combinations())
157 test_base.default_test_combinations(),
[all …]
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | shard_test.py | 20 from tensorflow.python.data.kernel_tests import test_base
27 class ShardTest(test_base.DatasetTestBase, parameterized.TestCase):
29 @combinations.generate(test_base.default_test_combinations())
34 @combinations.generate(test_base.default_test_combinations())
41 @combinations.generate(test_base.default_test_combinations())
46 @combinations.generate(test_base.default_test_combinations())
52 @combinations.generate(test_base.default_test_combinations())
58 @combinations.generate(test_base.default_test_combinations())
64 @combinations.generate(test_base.default_test_combinations())
70 @combinations.generate(test_base.default_test_combinations())
[all …]
|
| D | dataset_test.py | 26 from tensorflow.python.data.kernel_tests import test_base
50 class DatasetTest(test_base.DatasetTestBase, parameterized.TestCase):
52 @combinations.generate(test_base.default_test_combinations())
69 test_base.default_test_combinations(),
85 @combinations.generate(test_base.eager_only_combinations())
96 @combinations.generate(test_base.eager_only_combinations())
108 @combinations.generate(test_base.eager_only_combinations())
128 @combinations.generate(test_base.default_test_combinations())
133 @combinations.generate(test_base.default_test_combinations())
141 @combinations.generate(test_base.default_test_combinations())
[all …]
|
| D | as_numpy_iterator_test.py | 21 from tensorflow.python.data.kernel_tests import test_base
30 class AsNumpyIteratorTest(test_base.DatasetTestBase, parameterized.TestCase):
32 @combinations.generate(test_base.eager_only_combinations())
37 @combinations.generate(test_base.eager_only_combinations())
45 @combinations.generate(test_base.eager_only_combinations())
63 @combinations.generate(test_base.graph_only_combinations())
75 @combinations.generate(test_base.eager_only_combinations())
79 @combinations.generate(test_base.eager_only_combinations())
87 @combinations.generate(test_base.eager_only_combinations())
91 @combinations.generate(test_base.eager_only_combinations())
[all …]
|
| D | repeat_test.py | 21 from tensorflow.python.data.kernel_tests import test_base
29 class RepeatTest(test_base.DatasetTestBase, parameterized.TestCase):
32 combinations.times(test_base.default_test_combinations(),
43 @combinations.generate(test_base.default_test_combinations())
57 @combinations.generate(test_base.default_test_combinations())
71 @combinations.generate(test_base.default_test_combinations())
86 combinations.times(test_base.default_test_combinations(),
96 combinations.times(test_base.default_test_combinations(),
101 @combinations.generate(test_base.default_test_combinations())
111 combinations.times(test_base.default_test_combinations(),
[all …]
|
| D | padded_batch_test.py | 21 from tensorflow.python.data.kernel_tests import test_base
37 class PaddedBatchTest(test_base.DatasetTestBase, parameterized.TestCase):
41 test_base.default_test_combinations(),
84 @combinations.generate(test_base.default_test_combinations())
93 @combinations.generate(test_base.default_test_combinations())
101 @combinations.generate(test_base.default_test_combinations())
109 @combinations.generate(test_base.default_test_combinations())
115 @combinations.generate(test_base.default_test_combinations())
121 @combinations.generate(test_base.default_test_combinations())
134 @combinations.generate(test_base.default_test_combinations())
[all …]
|
| D | skip_test.py | 21 from tensorflow.python.data.kernel_tests import test_base
28 class SkipTest(test_base.DatasetTestBase, parameterized.TestCase):
31 combinations.times(test_base.default_test_combinations(),
44 @combinations.generate(test_base.default_test_combinations())
59 test_base.default_test_combinations(),
68 class SkipRandomAccessTest(test_base.DatasetTestBase, parameterized.TestCase):
71 combinations.times(test_base.default_test_combinations(),
79 combinations.times(test_base.default_test_combinations(),
87 combinations.times(test_base.default_test_combinations()))
94 combinations.times(test_base.default_test_combinations(),
[all …]
|
| D | range_test.py | 21 from tensorflow.python.data.kernel_tests import test_base
29 class RangeTest(test_base.DatasetTestBase, parameterized.TestCase):
33 test_base.default_test_combinations(),
46 test_base.default_test_combinations(),
59 test_base.default_test_combinations(),
74 test_base.default_test_combinations(),
87 test_base.default_test_combinations(),
102 test_base.default_test_combinations(),
115 test_base.default_test_combinations(),
130 test_base.default_test_combinations(),
[all …]
|
| D | prefetch_test.py | 23 from tensorflow.python.data.kernel_tests import test_base
30 class PrefetchTest(test_base.DatasetTestBase, parameterized.TestCase):
33 combinations.times(test_base.default_test_combinations(),
40 combinations.times(test_base.default_test_combinations(),
49 test_base.default_test_combinations(),
76 @combinations.generate(test_base.default_test_combinations())
90 combinations.times(test_base.default_test_combinations(),
96 class PrefetchRandomAccessTest(test_base.DatasetTestBase,
100 combinations.times(test_base.default_test_combinations(),
108 combinations.times(test_base.default_test_combinations(),
[all …]
|
| D | options_test.py | 24 from tensorflow.python.data.kernel_tests import test_base
33 class OptionsTest(test_base.DatasetTestBase, parameterized.TestCase):
41 @combinations.generate(test_base.default_test_combinations())
46 @combinations.generate(test_base.default_test_combinations())
52 @combinations.generate(test_base.default_test_combinations())
60 @combinations.generate(test_base.default_test_combinations())
74 @combinations.generate(test_base.default_test_combinations())
88 @combinations.generate(test_base.default_test_combinations())
101 @combinations.generate(test_base.default_test_combinations())
112 @combinations.generate(test_base.default_test_combinations())
[all …]
|
| D | from_generator_test.py | 21 from tensorflow.python.data.kernel_tests import test_base
36 class FromGeneratorTest(test_base.DatasetTestBase, parameterized.TestCase):
50 test_base.default_test_combinations(),
68 test_base.default_test_combinations(),
82 test_base.default_test_combinations(),
96 test_base.default_test_combinations(),
114 @combinations.generate(test_base.default_test_combinations())
198 @combinations.generate(test_base.default_test_combinations())
217 @combinations.generate(test_base.default_test_combinations())
229 @combinations.generate(test_base.default_test_combinations())
[all …]
|
| D | BUILD | 17 ":test_base",
36 ":test_base",
56 ":test_base",
80 ":test_base",
99 ":test_base",
112 ":test_base",
129 ":test_base",
158 ":test_base",
173 ":test_base",
188 ":test_base",
[all …]
|
| D | concatenate_test.py | 20 from tensorflow.python.data.kernel_tests import test_base
29 class ConcatenateTest(test_base.DatasetTestBase, parameterized.TestCase):
31 @combinations.generate(test_base.default_test_combinations())
65 @combinations.generate(test_base.default_test_combinations())
96 @combinations.generate(test_base.default_test_combinations())
113 @combinations.generate(test_base.default_test_combinations())
131 @combinations.generate(test_base.default_test_combinations())
147 @combinations.generate(test_base.default_test_combinations())
154 @combinations.generate(test_base.default_test_combinations())
175 combinations.times(test_base.default_test_combinations(),
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | image_grad_test.py | 17 from tensorflow.python.ops import image_grad_test_base as test_base unknown
20 ResizeNearestNeighborOpTest = test_base.ResizeNearestNeighborOpTestBase
21 ResizeBilinearOpTest = test_base.ResizeBilinearOpTestBase
22 ResizeBicubicOpTest = test_base.ResizeBicubicOpTestBase
23 ScaleAndTranslateOpTest = test_base.ScaleAndTranslateOpTestBase
24 CropAndResizeOpTest = test_base.CropAndResizeOpTestBase
25 RGBToHSVOpTest = test_base.RGBToHSVOpTestBase
|
| /external/tensorflow/tensorflow/python/data/util/ |
| D | options_test.py | 19 from tensorflow.python.data.kernel_tests import test_base
40 class OptionsTest(test_base.DatasetTestBase, parameterized.TestCase):
42 @combinations.generate(test_base.default_test_combinations())
47 @combinations.generate(test_base.default_test_combinations())
63 @combinations.generate(test_base.default_test_combinations())
77 @combinations.generate(test_base.default_test_combinations())
94 @combinations.generate(test_base.default_test_combinations())
106 @combinations.generate(test_base.default_test_combinations())
116 @combinations.generate(test_base.default_test_combinations())
122 @combinations.generate(test_base.default_test_combinations())
[all …]
|