Home
last modified time | relevance | path

Searched refs:input_dataset_params (Results 1 – 25 of 32) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Ddataset_test_base.h159 const std::vector<std::shared_ptr<DatasetParams>>& input_dataset_params() in input_dataset_params() function
225 BatchDatasetParams(T input_dataset_params, int64_t batch_size, in BatchDatasetParams() argument
235 input_dataset_params_.push_back(std::make_unique<T>(input_dataset_params)); in BatchDatasetParams()
238 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in BatchDatasetParams()
239 input_dataset_params.iterator_prefix()); in BatchDatasetParams()
261 MapDatasetParams(T input_dataset_params, std::vector<Tensor> other_arguments, in MapDatasetParams() argument
276 input_dataset_params_.push_back(std::make_unique<T>(input_dataset_params)); in MapDatasetParams()
278 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in MapDatasetParams()
279 input_dataset_params.iterator_prefix()); in MapDatasetParams()
336 TakeDatasetParams(T input_dataset_params, int count, in TakeDatasetParams() argument
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Drewrite_dataset_op_test.cc28 RewriteDatasetParams(T input_dataset_params, string rewrite_name, in RewriteDatasetParams() argument
35 input_dataset_params_.push_back(std::make_unique<T>(input_dataset_params)); in RewriteDatasetParams()
37 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in RewriteDatasetParams()
38 input_dataset_params.iterator_prefix()); in RewriteDatasetParams()
H A Doptimize_dataset_op_test.cc28 OptimizeDatasetParams(T input_dataset_params, string optimizations, in OptimizeDatasetParams() argument
37 input_dataset_params_.push_back(std::make_unique<T>(input_dataset_params)); in OptimizeDatasetParams()
39 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in OptimizeDatasetParams()
40 input_dataset_params.iterator_prefix()); in OptimizeDatasetParams()
H A Dzip_dataset_op_test.cc28 ZipDatasetParams(std::vector<T> input_dataset_params, in ZipDatasetParams() argument
35 for (auto& params : input_dataset_params) { in ZipDatasetParams()
40 name_utils::IteratorPrefix(input_dataset_params[0].dataset_type(), in ZipDatasetParams()
41 input_dataset_params[0].iterator_prefix()); in ZipDatasetParams()
H A Dreduce_dataset_op_test.cc27 ReduceDatasetParams(T input_dataset_params, std::vector<Tensor> initial_state, in ReduceDatasetParams() argument
44 input_dataset_params_.push_back(std::make_unique<T>(input_dataset_params)); in ReduceDatasetParams()
46 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in ReduceDatasetParams()
47 input_dataset_params.iterator_prefix()); in ReduceDatasetParams()
H A Dskip_dataset_op_test.cc25 SkipDatasetParams(T input_dataset_params, int64_t count, in SkipDatasetParams() argument
32 input_dataset_params_.push_back(std::make_unique<T>(input_dataset_params)); in SkipDatasetParams()
34 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in SkipDatasetParams()
35 input_dataset_params.iterator_prefix()); in SkipDatasetParams()
H A Dflat_map_dataset_op_test.cc25 FlatMapDatasetParams(T input_dataset_params, in FlatMapDatasetParams() argument
39 input_dataset_params_.push_back(std::make_unique<T>(input_dataset_params)); in FlatMapDatasetParams()
41 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in FlatMapDatasetParams()
42 input_dataset_params.iterator_prefix()); in FlatMapDatasetParams()
H A Dprefetch_dataset_op_test.cc28 PrefetchDatasetParams(T input_dataset_params, int64_t buffer_size, in PrefetchDatasetParams() argument
39 input_dataset_params_.push_back(std::make_unique<T>(input_dataset_params)); in PrefetchDatasetParams()
41 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in PrefetchDatasetParams()
42 input_dataset_params.iterator_prefix()); in PrefetchDatasetParams()
H A Dshard_dataset_op_test.cc25 ShardDatasetParams(T input_dataset_params, int64_t num_shards, int64_t index, in ShardDatasetParams() argument
34 input_dataset_params_.push_back(std::make_unique<T>(input_dataset_params)); in ShardDatasetParams()
36 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in ShardDatasetParams()
37 input_dataset_params.iterator_prefix()); in ShardDatasetParams()
H A Dfilter_dataset_op_test.cc25 FilterDatasetParams(T input_dataset_params, in FilterDatasetParams() argument
39 input_dataset_params_.push_back(std::make_unique<T>(input_dataset_params)); in FilterDatasetParams()
41 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in FilterDatasetParams()
42 input_dataset_params.iterator_prefix()); in FilterDatasetParams()
H A Dparallel_batch_dataset_op_test.cc29 ParallelBatchDatasetParams(T input_dataset_params, int64_t batch_size, in ParallelBatchDatasetParams() argument
42 input_dataset_params_.push_back(std::make_unique<T>(input_dataset_params)); in ParallelBatchDatasetParams()
45 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in ParallelBatchDatasetParams()
46 input_dataset_params.iterator_prefix()); in ParallelBatchDatasetParams()
H A Dcache_dataset_ops_test.cc33 CacheDatasetParams(T input_dataset_params, string filename, in CacheDatasetParams() argument
40 input_dataset_params_.push_back(std::make_unique<T>(input_dataset_params)); in CacheDatasetParams()
42 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in CacheDatasetParams()
43 input_dataset_params.iterator_prefix()); in CacheDatasetParams()
H A Drepeat_dataset_op_test.cc33 RepeatDatasetParams(T input_dataset_params, int64_t count, in RepeatDatasetParams() argument
40 input_dataset_params_.push_back(std::make_unique<T>(input_dataset_params)); in RepeatDatasetParams()
42 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in RepeatDatasetParams()
43 input_dataset_params.iterator_prefix()); in RepeatDatasetParams()
H A Dparallel_filter_dataset_op_test.cc32 T input_dataset_params, std::vector<Tensor> other_arguments, in ParallelFilterDatasetParams() argument
46 input_dataset_params_.push_back(std::make_unique<T>(input_dataset_params)); in ParallelFilterDatasetParams()
48 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in ParallelFilterDatasetParams()
49 input_dataset_params.iterator_prefix()); in ParallelFilterDatasetParams()
H A Dparallel_map_dataset_op_test.cc28 T input_dataset_params, std::vector<Tensor> other_arguments, in ParallelMapDatasetParams() argument
45 input_dataset_params_.push_back(std::make_unique<T>(input_dataset_params)); in ParallelMapDatasetParams()
50 input_dataset_params.dataset_type(), in ParallelMapDatasetParams()
51 input_dataset_params.iterator_prefix(), params); in ParallelMapDatasetParams()
H A Dinterleave_dataset_op_test.cc25 InterleaveDatasetParams(T input_dataset_params, in InterleaveDatasetParams() argument
42 input_dataset_params_.push_back(std::make_unique<T>(input_dataset_params)); in InterleaveDatasetParams()
44 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in InterleaveDatasetParams()
45 input_dataset_params.iterator_prefix()); in InterleaveDatasetParams()
H A Dshuffle_dataset_op_test.cc31 ShuffleDatasetParams(T input_dataset_params, int64_t buffer_size, in ShuffleDatasetParams() argument
44 input_dataset_params_.push_back(std::make_unique<T>(input_dataset_params)); in ShuffleDatasetParams()
46 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in ShuffleDatasetParams()
47 input_dataset_params.iterator_prefix()); in ShuffleDatasetParams()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Dsampling_dataset_op_test.cc28 SamplingDatasetParams(T input_dataset_params, float rate, in SamplingDatasetParams() argument
35 input_dataset_params_.push_back(std::make_unique<T>(input_dataset_params)); in SamplingDatasetParams()
37 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in SamplingDatasetParams()
38 input_dataset_params.iterator_prefix()); in SamplingDatasetParams()
H A Dassert_next_dataset_op_test.cc28 AssertNextDatasetParams(T input_dataset_params, in AssertNextDatasetParams() argument
36 input_dataset_params_.push_back(std::make_unique<T>(input_dataset_params)); in AssertNextDatasetParams()
38 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in AssertNextDatasetParams()
39 input_dataset_params.iterator_prefix()); in AssertNextDatasetParams()
H A Dunique_dataset_op_test.cc28 UniqueDatasetParams(T input_dataset_params, DataTypeVector output_dtypes, in UniqueDatasetParams() argument
32 input_dataset_params_.push_back(std::make_unique<T>(input_dataset_params)); in UniqueDatasetParams()
34 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in UniqueDatasetParams()
35 input_dataset_params.iterator_prefix()); in UniqueDatasetParams()
H A Dassert_prev_dataset_op_test.cc50 AssertPrevDatasetParams(T input_dataset_params, in AssertPrevDatasetParams() argument
58 input_dataset_params_.push_back(std::make_unique<T>(input_dataset_params)); in AssertPrevDatasetParams()
60 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in AssertPrevDatasetParams()
61 input_dataset_params.iterator_prefix()); in AssertPrevDatasetParams()
H A Dio_ops_test.cc37 SaveDatasetV2Params(T input_dataset_params, const tstring& path, in SaveDatasetV2Params() argument
52 input_dataset_params_.push_back(std::make_unique<T>(input_dataset_params)); in SaveDatasetV2Params()
54 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in SaveDatasetV2Params()
55 input_dataset_params.iterator_prefix()); in SaveDatasetV2Params()
H A Dmap_and_batch_dataset_op_test.cc27 T input_dataset_params, std::vector<Tensor> other_arguments, in MapAndBatchDatasetParams() argument
43 input_dataset_params_.push_back(std::make_unique<T>(input_dataset_params)); in MapAndBatchDatasetParams()
45 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in MapAndBatchDatasetParams()
46 input_dataset_params.iterator_prefix()); in MapAndBatchDatasetParams()
H A Dauto_shard_dataset_op_test.cc32 AutoShardDatasetParams(T input_dataset_params, int64_t num_workers, in AutoShardDatasetParams() argument
43 input_dataset_params_.push_back(std::make_unique<T>(input_dataset_params)); in AutoShardDatasetParams()
45 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in AutoShardDatasetParams()
46 input_dataset_params.iterator_prefix()); in AutoShardDatasetParams()
H A Dparallel_interleave_dataset_op_test.cc29 T input_dataset_params, std::vector<Tensor> other_arguments, in ParallelInterleaveDatasetParams() argument
47 input_dataset_params_.push_back(std::make_unique<T>(input_dataset_params)); in ParallelInterleaveDatasetParams()
52 input_dataset_params.dataset_type(), in ParallelInterleaveDatasetParams()
53 input_dataset_params.iterator_prefix(), params); in ParallelInterleaveDatasetParams()

12