Home
last modified time | relevance | path

Searched defs:dataset_params (Results 1 – 25 of 32) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dassert_next_dataset_op_test.cc145 auto dataset_params = AssertNextDatasetParams1(); in ITERATOR_GET_NEXT_TEST_P() local
151 auto dataset_params = AssertNextDatasetParams1(); in TEST_F() local
158 auto dataset_params = AssertNextDatasetParams1(); in TEST_F() local
164 auto dataset_params = AssertNextDatasetParams1(); in TEST_F() local
170 auto dataset_params = AssertNextDatasetParams1(); in TEST_F() local
176 auto dataset_params = AssertNextDatasetParams1(); in TEST_F() local
182 auto dataset_params = AssertNextDatasetParams1(); in TEST_F() local
188 auto dataset_params = AssertNextDatasetParams1(); in TEST_F() local
211 auto dataset_params = InvalidAssertNextDatasetParams(); in ITERATOR_SAVE_AND_RESTORE_TEST_P() local
217 auto dataset_params = ShortAssertNextDatasetParams(); in TEST_F() local
Ddirected_interleave_dataset_op_test.cc258 auto dataset_params = AlternateInputsParams(); in ITERATOR_GET_NEXT_TEST_P() local
264 auto dataset_params = AlternateInputsParams(); in TEST_F() local
271 auto dataset_params = AlternateInputsParams(); in TEST_F() local
277 auto dataset_params = AlternateInputsParams(); in TEST_F() local
283 auto dataset_params = AlternateInputsParams(); in TEST_F() local
289 auto dataset_params = AlternateInputsParams(); in TEST_F() local
295 auto dataset_params = AlternateInputsParams(); in TEST_F() local
301 auto dataset_params = AlternateInputsParams(); in TEST_F() local
345 for (auto& dataset_params : invalid_params_vec) { in ITERATOR_SAVE_AND_RESTORE_TEST_P() local
352 auto dataset_params = InvalidSelectorValues(); in TEST_F() local
Dparallel_interleave_dataset_op_test.cc410 auto dataset_params = ParallelInterleaveDatasetParams1(); in ITERATOR_GET_NEXT_TEST_P() local
416 auto dataset_params = ParallelInterleaveDatasetParams1(); in TEST_F() local
425 auto dataset_params = ParallelInterleaveDatasetParams1(); in TEST_F() local
431 auto dataset_params = ParallelInterleaveDatasetParams1(); in TEST_F() local
455 auto dataset_params = ParallelInterleaveDatasetParams1(); in DATASET_CARDINALITY_TEST_P() local
461 auto dataset_params = ParallelInterleaveDatasetParams1(); in TEST_F() local
467 auto dataset_params = ParallelInterleaveDatasetParams1(); in TEST_F() local
520 for (auto& dataset_params : invalid_params) { in ITERATOR_SAVE_AND_RESTORE_TEST_P() local
Dmap_and_batch_dataset_op_test.cc273 auto dataset_params = MapAndBatchDatasetParams1(); in ITERATOR_GET_NEXT_TEST_P() local
279 auto dataset_params = MapAndBatchDatasetParams1(); in TEST_F() local
286 auto dataset_params = MapAndBatchDatasetParams1(); in TEST_F() local
336 auto dataset_params = MapAndBatchDatasetParams1(); in DATASET_CARDINALITY_TEST_P() local
368 auto dataset_params = MapAndBatchDatasetParams1(); in ITERATOR_OUTPUT_SHAPES_TEST_P() local
411 auto dataset_params = InvalidBatchSizeMapAndBatchDatasetParams(); in ITERATOR_SAVE_AND_RESTORE_TEST_P() local
417 auto dataset_params = InvalidNumParallelCallsMapAndBatchDatasetParams(); in TEST_F() local
Dunique_dataset_op_test.cc182 auto dataset_params = NormalCaseParams(); in ITERATOR_GET_NEXT_TEST_P() local
188 auto dataset_params = NormalCaseParams(); in TEST_F() local
195 auto dataset_params = NormalCaseParams(); in TEST_F() local
201 auto dataset_params = NormalCaseParams(); in TEST_F() local
241 auto dataset_params = NormalCaseParams(); in ITERATOR_OUTPUT_SHAPES_TEST_P() local
269 auto dataset_params = GetParam(); in TEST_P() local
/external/tensorflow/tensorflow/core/kernels/data/
Dmap_dataset_op_test.cc109 auto dataset_params = MapDatasetParams1(); in ITERATOR_GET_NEXT_TEST_P() local
115 auto dataset_params = MapDatasetParams1(); in TEST_F() local
122 auto dataset_params = MapDatasetParams1(); in TEST_F() local
128 auto dataset_params = MapDatasetParams1(); in TEST_F() local
146 auto dataset_params = MapDatasetParams1(); in DATASET_CARDINALITY_TEST_P() local
152 auto dataset_params = MapDatasetParams1(); in TEST_F() local
158 auto dataset_params = MapDatasetParams1(); in TEST_F() local
Dflat_map_dataset_op_test.cc134 auto dataset_params = FlatMapDatasetParams1(); in ITERATOR_GET_NEXT_TEST_P() local
140 auto dataset_params = FlatMapDatasetParams1(); in TEST_F() local
147 auto dataset_params = FlatMapDatasetParams1(); in TEST_F() local
153 auto dataset_params = FlatMapDatasetParams1(); in TEST_F() local
159 auto dataset_params = FlatMapDatasetParams1(); in TEST_F() local
165 auto dataset_params = FlatMapDatasetParams1(); in TEST_F() local
171 auto dataset_params = FlatMapDatasetParams1(); in TEST_F() local
177 auto dataset_params = FlatMapDatasetParams1(); in TEST_F() local
197 auto dataset_params = InvalidFlatMapDatasetParams(); in ITERATOR_SAVE_AND_RESTORE_TEST_P() local
Dshard_dataset_op_test.cc227 auto dataset_params = ShardDatasetParams1(); in ITERATOR_GET_NEXT_TEST_P() local
233 auto dataset_params = ShardDatasetParams1(); in TEST_F() local
240 auto dataset_params = ShardDatasetParams1(); in TEST_F() local
246 auto dataset_params = ShardDatasetParams1(); in TEST_F() local
272 auto dataset_params = ShardDatasetParams1(); in DATASET_CARDINALITY_TEST_P() local
278 auto dataset_params = ShardDatasetParams1(); in TEST_F() local
284 auto dataset_params = ShardDatasetParams1(); in TEST_F() local
320 auto dataset_params = InvalidShardDatasetParamsWithNoElemForEachShard(); in ITERATOR_SAVE_AND_RESTORE_TEST_P() local
334 for (const auto& dataset_params : invalid_dataset_params) { in TEST_F() local
Dprefetch_dataset_op_test.cc234 auto dataset_params = PrefetchDatasetParams1(); in ITERATOR_GET_NEXT_TEST_P() local
240 auto dataset_params = PrefetchDatasetParams1(); in TEST_F() local
247 auto dataset_params = PrefetchDatasetParams1(); in TEST_F() local
253 auto dataset_params = PrefetchDatasetParams1(); in TEST_F() local
275 auto dataset_params = PrefetchDatasetParams1(); in DATASET_CARDINALITY_TEST_P() local
281 auto dataset_params = PrefetchDatasetParams1(); in TEST_F() local
287 auto dataset_params = PrefetchDatasetParams1(); in TEST_F() local
331 auto dataset_params = InvalidBufferSizePrefetchDatasetParams(); in ITERATOR_SAVE_AND_RESTORE_TEST_P() local
Dtext_line_dataset_op_test.cc168 auto dataset_params = TextLineDatasetParams1(); in ITERATOR_GET_NEXT_TEST_P() local
174 auto dataset_params = TextLineDatasetParams1(); in TEST_F() local
181 auto dataset_params = TextLineDatasetParams1(); in TEST_F() local
187 auto dataset_params = TextLineDatasetParams1(); in TEST_F() local
193 auto dataset_params = TextLineDatasetParams1(); in TEST_F() local
199 auto dataset_params = TextLineDatasetParams1(); in TEST_F() local
205 auto dataset_params = TextLineDatasetParams1(); in TEST_F() local
211 auto dataset_params = TextLineDatasetParams1(); in TEST_F() local
Dtf_record_dataset_op_test.cc196 auto dataset_params = TFRecordDatasetParams1(); in ITERATOR_SKIP_TEST_P() local
202 auto dataset_params = TFRecordDatasetParams1(); in TEST_F() local
209 auto dataset_params = TFRecordDatasetParams1(); in TEST_F() local
215 auto dataset_params = TFRecordDatasetParams1(); in TEST_F() local
221 auto dataset_params = TFRecordDatasetParams1(); in TEST_F() local
227 auto dataset_params = TFRecordDatasetParams1(); in TEST_F() local
233 auto dataset_params = TFRecordDatasetParams1(); in TEST_F() local
239 auto dataset_params = TFRecordDatasetParams1(); in TEST_F() local
Dfixed_length_record_dataset_op_test.cc192 auto dataset_params = FixedLengthRecordDatasetParams1(); in ITERATOR_GET_NEXT_TEST_P() local
198 auto dataset_params = FixedLengthRecordDatasetParams1(); in TEST_F() local
207 auto dataset_params = FixedLengthRecordDatasetParams1(); in TEST_F() local
213 auto dataset_params = FixedLengthRecordDatasetParams1(); in TEST_F() local
219 auto dataset_params = FixedLengthRecordDatasetParams1(); in TEST_F() local
225 auto dataset_params = FixedLengthRecordDatasetParams1(); in TEST_F() local
231 auto dataset_params = FixedLengthRecordDatasetParams1(); in TEST_F() local
237 auto dataset_params = FixedLengthRecordDatasetParams1(); in TEST_F() local
Dskip_dataset_op_test.cc135 auto dataset_params = SkipDatasetParams1(); in ITERATOR_GET_NEXT_TEST_P() local
141 auto dataset_params = SkipDatasetParams1(); in TEST_F() local
148 auto dataset_params = SkipDatasetParams1(); in TEST_F() local
154 auto dataset_params = SkipDatasetParams1(); in TEST_F() local
176 auto dataset_params = SkipDatasetParams1(); in DATASET_CARDINALITY_TEST_P() local
182 auto dataset_params = SkipDatasetParams1(); in TEST_F() local
Dtensor_dataset_op_test.cc151 auto dataset_params = PlainTensorDatasetParams(); in TEST_F() local
158 auto dataset_params = PlainTensorDatasetParams(); in TEST_F() local
164 auto dataset_params = PlainTensorDatasetParams(); in TEST_F() local
170 auto dataset_params = PlainTensorDatasetParams(); in TEST_F() local
176 auto dataset_params = PlainTensorDatasetParams(); in TEST_F() local
182 auto dataset_params = PlainTensorDatasetParams(); in TEST_F() local
188 auto dataset_params = PlainTensorDatasetParams(); in TEST_F() local
194 auto dataset_params = PlainTensorDatasetParams(); in TEST_F() local
Dparallel_map_dataset_op_test.cc303 auto dataset_params = ParallelMapDatasetParams1(); in ITERATOR_GET_NEXT_TEST_P() local
309 auto dataset_params = ParallelMapDatasetParams1(); in TEST_F() local
318 auto dataset_params = ParallelMapDatasetParams1(); in TEST_F() local
324 auto dataset_params = ParallelMapDatasetParams1(); in TEST_F() local
349 auto dataset_params = ParallelMapDatasetParams1(); in DATASET_CARDINALITY_TEST_P() local
355 auto dataset_params = ParallelMapDatasetParams1(); in TEST_F() local
361 auto dataset_params = ParallelMapDatasetParams1(); in TEST_F() local
410 auto dataset_params = ParallelMapDatasetParamsWithInvalidNumParallelCalls(); in ITERATOR_SAVE_AND_RESTORE_TEST_P() local
Dwindow_dataset_op_test.cc236 T dataset_params; member
331 auto dataset_params = WindowDatasetParams1(); in TEST_F() local
338 auto dataset_params = WindowDatasetParams1(); in TEST_F() local
344 auto dataset_params = WindowDatasetParams1(); in TEST_F() local
350 auto dataset_params = WindowDatasetParams1(); in TEST_F() local
383 auto dataset_params = WindowDatasetParams1(); in DATASET_CARDINALITY_TEST_P() local
389 auto dataset_params = WindowDatasetParams1(); in TEST_F() local
395 auto dataset_params = WindowDatasetParams1(); in TEST_F() local
403 T dataset_params; member
533 for (const auto& dataset_params : dataset_params_vec) { in TEST_F() local
Dfilter_dataset_op_test.cc184 auto dataset_params = FilterDatasetParams1(); in ITERATOR_GET_NEXT_TEST_P() local
190 auto dataset_params = FilterDatasetParams1(); in TEST_F() local
197 auto dataset_params = FilterDatasetParams1(); in TEST_F() local
224 auto dataset_params = FilterDatasetParams1(); in DATASET_CARDINALITY_TEST_P() local
241 auto dataset_params = FilterDatasetParams1(); in ITERATOR_OUTPUT_SHAPES_TEST_P() local
266 auto dataset_params = GetParam(); in TEST_P() local
Ddataset_test_base.h375 : dataset_params(std::move(dataset_params)), in dataset_params() function
379 T dataset_params; member
389 : dataset_params(std::move(dataset_params)), in dataset_params() function
396 T dataset_params; member
406 T dataset_params; member
412 T dataset_params; member
418 T dataset_params; member
424 T dataset_params; member
430 T dataset_params; member
436 T dataset_params; member
[all …]
Dparallel_interleave_dataset_op_test.cc608 auto dataset_params = ParallelInterleaveDatasetParams1(); in ITERATOR_GET_NEXT_TEST_P() local
614 auto dataset_params = ParallelInterleaveDatasetParams1(); in TEST_F() local
623 auto dataset_params = ParallelInterleaveDatasetParams1(); in TEST_F() local
629 auto dataset_params = ParallelInterleaveDatasetParams1(); in TEST_F() local
663 auto dataset_params = ParallelInterleaveDatasetParams1(); in DATASET_CARDINALITY_TEST_P() local
669 auto dataset_params = ParallelInterleaveDatasetParams1(); in TEST_F() local
675 auto dataset_params = ParallelInterleaveDatasetParams1(); in TEST_F() local
771 for (auto& dataset_params : invalid_params) { in ITERATOR_SAVE_AND_RESTORE_TEST_P() local
Dshuffle_dataset_op_test.cc245 T dataset_params; member
399 auto dataset_params = ShuffleDatasetParams1(); in DATASET_TYPE_STRING_TEST_P() local
405 auto dataset_params = ShuffleDatasetParams1(); in TEST_F() local
433 auto dataset_params = ShuffleDatasetParams1(); in DATASET_CARDINALITY_TEST_P() local
439 auto dataset_params = ShuffleDatasetParams1(); in TEST_F() local
445 auto dataset_params = ShuffleDatasetParams1(); in TEST_F() local
453 T dataset_params; member
557 for (const auto& dataset_params : dataset_params_vec) { in TEST_F() local
Dtake_dataset_op_test.cc82 auto dataset_params = TakeLessTakeDatasetParams(); in ITERATOR_GET_NEXT_TEST_P() local
88 auto dataset_params = TakeLessTakeDatasetParams(); in TEST_F() local
95 auto dataset_params = TakeLessTakeDatasetParams(); in TEST_F() local
130 auto dataset_params = TakeLessTakeDatasetParams(); in DATASET_CARDINALITY_TEST_P() local
Dcache_dataset_ops_test.cc69 Status Initialize(const DatasetParams& dataset_params) { in Initialize()
203 auto dataset_params = CacheDatasetParams1(); in TEST_F() local
209 auto dataset_params = CacheDatasetParams1(); in TEST_F() local
216 auto dataset_params = CacheDatasetParams1(); in TEST_F() local
252 auto dataset_params = CacheDatasetParams1(); in DATASET_CARDINALITY_TEST_P() local
274 auto dataset_params = CacheDatasetParams1(); in ITERATOR_OUTPUT_SHAPES_TEST_P() local
Dconcatenate_dataset_op_test.cc108 auto dataset_params = DifferentDtypeConcatenateDatasetParams(); in ITERATOR_GET_NEXT_TEST_P() local
115 auto dataset_params = SameShapeConcatenateDatasetParams(); in TEST_F() local
121 auto dataset_params = SameShapeConcatenateDatasetParams(); in TEST_F() local
194 auto dataset_params = SameShapeConcatenateDatasetParams(); in ITERATOR_OUTPUT_SHAPES_TEST_P() local
Dsparse_tensor_slice_dataset_op_test.cc107 T dataset_params; member
189 auto dataset_params = TwoDimsSparseTensorSliceDatasetParams(); in TEST_F() local
195 auto dataset_params = TwoDimsSparseTensorSliceDatasetParams(); in TEST_F() local
297 auto dataset_params = TwoDimsSparseTensorSliceDatasetParams(); in ITERATOR_OUTPUT_SHAPES_TEST_P() local
305 T dataset_params; member
Dinterleave_dataset_op_test.cc339 auto dataset_params = InterleaveDatasetParams1(); in ITERATOR_GET_NEXT_TEST_P() local
345 auto dataset_params = InterleaveDatasetParams1(); in TEST_F() local
457 auto dataset_params = InterleaveDatasetParams1(); in ITERATOR_OUTPUT_SHAPES_TEST_P() local
511 auto dataset_params = InterleaveDatasetParamsWithInvalidCycleLength(); in ITERATOR_SAVE_AND_RESTORE_TEST_P() local
517 auto dataset_params = InterleaveDatasetParamsWithInvalidBlockLength(); in TEST_F() local

12