Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/kernels/data/
Doptimize_dataset_op_test.cc28 OptimizeDatasetParams(T input_dataset_params, string optimizations, in OptimizeDatasetParams() argument
37 input_dataset_params_.push_back(absl::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()
Dreduce_dataset_op_test.cc27 ReduceDatasetParams(T input_dataset_params, std::vector<Tensor> initial_state, in ReduceDatasetParams() argument
44 input_dataset_params_.push_back(absl::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()
Dskip_dataset_op_test.cc25 SkipDatasetParams(T input_dataset_params, int64 count, in SkipDatasetParams() argument
32 input_dataset_params_.push_back(absl::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()
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()
Dflat_map_dataset_op_test.cc25 FlatMapDatasetParams(T input_dataset_params, in FlatMapDatasetParams() argument
39 input_dataset_params_.push_back(absl::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()
Dprefetch_dataset_op_test.cc28 PrefetchDatasetParams(T input_dataset_params, int64 buffer_size, in PrefetchDatasetParams() argument
39 input_dataset_params_.push_back(absl::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()
Ddataset_test_base.h151 const std::vector<std::shared_ptr<DatasetParams>>& input_dataset_params() in input_dataset_params() function
208 BatchDatasetParams(T input_dataset_params, int64 batch_size, in BatchDatasetParams() argument
218 input_dataset_params_.push_back(std::make_unique<T>(input_dataset_params)); in BatchDatasetParams()
221 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in BatchDatasetParams()
222 input_dataset_params.iterator_prefix()); in BatchDatasetParams()
244 MapDatasetParams(T input_dataset_params, std::vector<Tensor> other_arguments, in MapDatasetParams() argument
259 input_dataset_params_.push_back(absl::make_unique<T>(input_dataset_params)); in MapDatasetParams()
261 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in MapDatasetParams()
262 input_dataset_params.iterator_prefix()); in MapDatasetParams()
317 TakeDatasetParams(T input_dataset_params, int count, in TakeDatasetParams() argument
[all …]
Dfilter_dataset_op_test.cc25 FilterDatasetParams(T input_dataset_params, in FilterDatasetParams() argument
39 input_dataset_params_.push_back(absl::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()
Dshard_dataset_op_test.cc25 ShardDatasetParams(T input_dataset_params, int64 num_shards, int64 index, in ShardDatasetParams() argument
34 input_dataset_params_.push_back(absl::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()
Dparallel_batch_dataset_op_test.cc29 ParallelBatchDatasetParams(T input_dataset_params, int64 batch_size, in ParallelBatchDatasetParams() argument
39 input_dataset_params_.push_back(std::make_unique<T>(input_dataset_params)); in ParallelBatchDatasetParams()
42 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in ParallelBatchDatasetParams()
43 input_dataset_params.iterator_prefix()); in ParallelBatchDatasetParams()
Drepeat_dataset_op_test.cc29 RepeatDatasetParams(T input_dataset_params, int64 count, in RepeatDatasetParams() argument
36 input_dataset_params_.push_back(absl::make_unique<T>(input_dataset_params)); in RepeatDatasetParams()
38 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in RepeatDatasetParams()
39 input_dataset_params.iterator_prefix()); in RepeatDatasetParams()
Dcache_dataset_ops_test.cc29 CacheDatasetParams(T input_dataset_params, string filename, in CacheDatasetParams() argument
36 input_dataset_params_.push_back(absl::make_unique<T>(input_dataset_params)); in CacheDatasetParams()
38 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in CacheDatasetParams()
39 input_dataset_params.iterator_prefix()); in CacheDatasetParams()
Dparallel_map_dataset_op_test.cc28 T input_dataset_params, std::vector<Tensor> other_arguments, in ParallelMapDatasetParams() argument
45 input_dataset_params_.push_back(absl::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()
Dinterleave_dataset_op_test.cc25 InterleaveDatasetParams(T input_dataset_params, in InterleaveDatasetParams() argument
42 input_dataset_params_.push_back(absl::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()
Dshuffle_dataset_op_test.cc27 ShuffleDatasetParams(T input_dataset_params, int64 buffer_size, int64 seed, in ShuffleDatasetParams() argument
39 input_dataset_params_.push_back(absl::make_unique<T>(input_dataset_params)); in ShuffleDatasetParams()
41 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in ShuffleDatasetParams()
42 input_dataset_params.iterator_prefix()); in ShuffleDatasetParams()
Dwindow_dataset_op_test.cc26 WindowDatasetParams(T input_dataset_params, int64 size, int64 shift, in WindowDatasetParams() argument
37 input_dataset_params_.push_back(absl::make_unique<T>(input_dataset_params)); in WindowDatasetParams()
39 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in WindowDatasetParams()
40 input_dataset_params.iterator_prefix()); in WindowDatasetParams()
Dpadded_batch_dataset_op_test.cc28 PaddedBatchDatasetParams(T input_dataset_params, int64 batch_size, in PaddedBatchDatasetParams() argument
43 input_dataset_params_.push_back(absl::make_unique<T>(input_dataset_params)); in PaddedBatchDatasetParams()
46 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in PaddedBatchDatasetParams()
47 input_dataset_params.iterator_prefix()); in PaddedBatchDatasetParams()
Dparallel_interleave_dataset_op_test.cc27 T input_dataset_params, std::vector<Tensor> other_arguments, in ParallelInterleaveDatasetParams() argument
47 input_dataset_params_.push_back(absl::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()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dauto_shard_dataset_op_test.cc27 AutoShardDatasetParams(T input_dataset_params, int64 num_workers, int64 index, in AutoShardDatasetParams() argument
38 input_dataset_params_.push_back(absl::make_unique<T>(input_dataset_params)); in AutoShardDatasetParams()
40 name_utils::IteratorPrefix(input_dataset_params.dataset_type(), in AutoShardDatasetParams()
41 input_dataset_params.iterator_prefix()); in AutoShardDatasetParams()
Dsampling_dataset_op_test.cc28 SamplingDatasetParams(T input_dataset_params, float rate, in SamplingDatasetParams() argument
35 input_dataset_params_.push_back(absl::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()
Dassert_next_dataset_op_test.cc28 AssertNextDatasetParams(T input_dataset_params, in AssertNextDatasetParams() argument
36 input_dataset_params_.push_back(absl::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()
Dunique_dataset_op_test.cc28 UniqueDatasetParams(T input_dataset_params, DataTypeVector output_dtypes, in UniqueDatasetParams() argument
32 input_dataset_params_.push_back(absl::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()
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(absl::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()
Dparallel_interleave_dataset_op_test.cc29 T input_dataset_params, std::vector<Tensor> other_arguments, in ParallelInterleaveDatasetParams() argument
47 input_dataset_params_.push_back(absl::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()
Ddirected_interleave_dataset_op_test.cc36 for (auto input_dataset_params : input_dataset_params_vec) { in DirectedInterleaveDatasetParams() local
38 absl::make_unique<T>(input_dataset_params)); in DirectedInterleaveDatasetParams()

12