Home
last modified time | relevance | path

Searched refs:iterator_prefix (Results 1 – 25 of 40) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/data/
Dskip_dataset_op_test.cc35 input_dataset_params.iterator_prefix()); in SkipDatasetParams()
192 SkipDatasetParams1().iterator_prefix())}, in IteratorPrefixTestCases()
195 "FiniteSkip", SkipDatasetParams2().iterator_prefix())}, in IteratorPrefixTestCases()
198 "FiniteSkip", SkipDatasetParams3().iterator_prefix())}, in IteratorPrefixTestCases()
201 "FiniteSkip", SkipDatasetParams4().iterator_prefix())}, in IteratorPrefixTestCases()
204 "EmptySkip", SkipDatasetParams5().iterator_prefix())}}; in IteratorPrefixTestCases()
Dtake_dataset_op_test.cc154 "FiniteTake", TakeLessTakeDatasetParams().iterator_prefix())}, in ITERATOR_OUTPUT_SHAPES_TEST_P()
157 "FiniteTake", TakeMoreTakeDatasetParams().iterator_prefix())}, in ITERATOR_OUTPUT_SHAPES_TEST_P()
160 "FiniteTake", TakeAllTakeDatasetParams().iterator_prefix())}, in ITERATOR_OUTPUT_SHAPES_TEST_P()
163 "EmptyTake", TakeNothingTakeDatasetParams().iterator_prefix())}}; in ITERATOR_OUTPUT_SHAPES_TEST_P()
Drepeat_dataset_op_test.cc39 input_dataset_params.iterator_prefix()); in RepeatDatasetParams()
265 "FiniteRepeat", FiniteRepeatDatasetParams().iterator_prefix())}, in ITERATOR_OUTPUT_SHAPES_TEST_P()
268 "EmptyRepeat", EmptyRepeatDatasetParams().iterator_prefix())}, in ITERATOR_OUTPUT_SHAPES_TEST_P()
271 "ForeverRepeat", ForeverRepeatDatasetParams().iterator_prefix())}}; in ITERATOR_OUTPUT_SHAPES_TEST_P()
327 test_case.dataset_params.iterator_prefix(), in TEST_P()
Dcache_dataset_ops_test.cc39 input_dataset_params.iterator_prefix()); in CacheDatasetParams()
186 test_case.dataset_params.iterator_prefix(), &iterator_)); in TEST_P()
280 CacheDatasetOp::kDatasetType, dataset_params.iterator_prefix(), in ITERATOR_OUTPUT_SHAPES_TEST_P()
326 test_case.dataset_params.iterator_prefix(), &iterator_)); in TEST_P()
340 test_case.dataset_params.iterator_prefix(), in TEST_P()
Ddataset_utils.cc711 Status ReadBatch(int64 batch_size, const string& iterator_prefix, in ReadBatch() argument
716 FullName(iterator_prefix, in ReadBatch()
723 FullName(iterator_prefix, in ReadBatch()
746 const string& iterator_prefix, const string& batch_prefix, in WriteBatch() argument
749 FullName(iterator_prefix, in WriteBatch()
758 FullName(iterator_prefix, in WriteBatch()
763 FullName(iterator_prefix, in WriteBatch()
771 Status ReadStatus(const string& iterator_prefix, const string& prefix, in ReadStatus() argument
775 FullName(iterator_prefix, strings::StrCat(prefix, "_", kCode)), in ReadStatus()
782 FullName(iterator_prefix, strings::StrCat(prefix, "_", kMessage)), in ReadStatus()
[all …]
Dwindow_dataset_op_test.cc40 input_dataset_params.iterator_prefix()); in WindowDatasetParams()
306 test_case.dataset_params.iterator_prefix(), in TEST_P()
398 WindowDatasetOp::kDatasetType, dataset_params.iterator_prefix()))); in TEST_F()
482 test_case.dataset_params.iterator_prefix(), in TEST_P()
500 test_case.dataset_params.iterator_prefix(), in TEST_P()
Ddataset_utils.h304 Status ReadBatch(int64 batch_size, const string& iterator_prefix,
310 const string& iterator_prefix, const string& batch_prefix,
314 Status ReadStatus(const string& iterator_prefix, const string& prefix,
318 Status WriteStatus(const string& iterator_prefix, const string& prefix,
Dshuffle_dataset_op_test.cc42 input_dataset_params.iterator_prefix()); in ShuffleDatasetParams()
346 test_case.dataset_params.iterator_prefix(), &iterator_)); in TEST_P()
448 ShuffleDatasetOp::kDatasetType, dataset_params.iterator_prefix()))); in TEST_F()
532 test_case.dataset_params.iterator_prefix(), in TEST_P()
Dzip_dataset_op_test.cc41 input_dataset_params[0].iterator_prefix()); in ZipDatasetParams()
181 ZipDatasetOp::kDatasetType, dataset_params.iterator_prefix()))); in ITERATOR_OUTPUT_SHAPES_TEST_P()
Ddataset_test_base.h149 string iterator_prefix() const { return iterator_prefix_; } in iterator_prefix() function
222 input_dataset_params.iterator_prefix()); in BatchDatasetParams()
262 input_dataset_params.iterator_prefix()); in MapDatasetParams()
327 input_dataset_params.iterator_prefix()); in TakeDatasetParams()
359 input_dataset_params_0.iterator_prefix()); in ConcatenateDatasetParams()
632 const std::string& iterator_prefix,
637 const std::string& iterator_prefix,
992 test_case.dataset_params.iterator_prefix(), \
Dflat_map_dataset_op_test.cc42 input_dataset_params.iterator_prefix()); in FlatMapDatasetParams()
180 FlatMapDatasetOp::kDatasetType, dataset_params.iterator_prefix()))); in TEST_F()
Dprefetch_dataset_op_test.cc42 input_dataset_params.iterator_prefix()); in PrefetchDatasetParams()
290 PrefetchDatasetOp::kDatasetType, dataset_params.iterator_prefix()))); in TEST_F()
Dfilter_dataset_op_test.cc42 input_dataset_params.iterator_prefix()); in FilterDatasetParams()
244 FilterDatasetOp::kDatasetType, dataset_params.iterator_prefix()))); in ITERATOR_OUTPUT_SHAPES_TEST_P()
Dshard_dataset_op_test.cc37 input_dataset_params.iterator_prefix()); in ShardDatasetParams()
287 ShardDatasetOp::kDatasetType, dataset_params.iterator_prefix()))); in TEST_F()
Dparallel_batch_dataset_op_test.cc43 input_dataset_params.iterator_prefix()); in ParallelBatchDatasetParams()
352 parallel_batch_dataset_params.iterator_prefix(), params))); in ITERATOR_OUTPUT_SHAPES_TEST_P()
Ddataset_test_base.cc664 dataset->dataset(), iterator_ctx.get(), params.iterator_prefix(), in CheckSplitProviderShardedIteration()
725 const std::string& iterator_prefix, in CheckIteratorSaveAndRestore() argument
730 iterator_prefix, &iterator)); in CheckIteratorSaveAndRestore()
742 TF_EXPECT_OK(RestoreIterator(iterator_ctx, &reader, iterator_prefix, in CheckIteratorSaveAndRestore()
759 const std::string& iterator_prefix, in CheckIteratorSaveAndRestore() argument
763 iterator_prefix, expected_outputs, in CheckIteratorSaveAndRestore()
779 dataset_params.iterator_prefix(), &iterator_)); in Initialize()
869 iterator_ctx.get(), /*parent=*/nullptr, dataset_params.iterator_prefix(), in MakeIterator()
Doptimize_dataset_op_test.cc40 input_dataset_params.iterator_prefix()); in OptimizeDatasetParams()
Dtensor_dataset_op_test.cc197 "FromTensor", dataset_params.iterator_prefix()))); in TEST_F()
244 test_case.dataset_params.iterator_prefix(), in TEST_P()
Dparallel_map_dataset_op_test.cc51 input_dataset_params.iterator_prefix(), params); in ParallelMapDatasetParams()
367 dataset_params.iterator_prefix(), params))); in TEST_F()
Dsparse_tensor_slice_dataset_op_test.cc300 kDatasetType, dataset_params.iterator_prefix()))); in ITERATOR_OUTPUT_SHAPES_TEST_P()
407 test_case.dataset_params.iterator_prefix(), in TEST_P()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dsampling_dataset_op_test.cc38 input_dataset_params.iterator_prefix()); in SamplingDatasetParams()
191 TenPercentSampleParams().iterator_prefix())}}; in ITERATOR_OUTPUT_SHAPES_TEST_P()
Dassert_next_dataset_op_test.cc39 input_dataset_params.iterator_prefix()); in AssertNextDatasetParams()
191 AssertNextDatasetOp::kDatasetType, dataset_params.iterator_prefix()))); in TEST_F()
Dunique_dataset_op_test.cc35 input_dataset_params.iterator_prefix()); in UniqueDatasetParams()
244 UniqueDatasetOp::kDatasetType, dataset_params.iterator_prefix()))); in ITERATOR_OUTPUT_SHAPES_TEST_P()
Ddirected_interleave_dataset_op_test.cc44 input_dataset_params_vec[0].iterator_prefix()); in DirectedInterleaveDatasetParams()
305 dataset_params.iterator_prefix()))); in TEST_F()
Dmap_and_batch_dataset_op_test.cc46 input_dataset_params.iterator_prefix()); in MapAndBatchDatasetParams()
371 MapAndBatchDatasetOp::kDatasetType, dataset_params.iterator_prefix()))); in ITERATOR_OUTPUT_SHAPES_TEST_P()

12