Home
last modified time | relevance | path

Searched defs:input_dataset_params (Results 1 – 25 of 31) 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()
261 MapDatasetParams(T input_dataset_params, std::vector<Tensor> other_arguments, in MapDatasetParams()
336 TakeDatasetParams(T input_dataset_params, int count, in TakeDatasetParams()
395 OptionsDatasetParams(T input_dataset_params, const string& serialized_options, in OptionsDatasetParams()
/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()
H A Doptimize_dataset_op_test.cc28 OptimizeDatasetParams(T input_dataset_params, string optimizations, in OptimizeDatasetParams()
H A Dget_options_op_test.cc35 GetOptionsParams(T input_dataset_params, DataTypeVector output_dtypes, in GetOptionsParams()
H A Dreduce_dataset_op_test.cc27 ReduceDatasetParams(T input_dataset_params, std::vector<Tensor> initial_state, in ReduceDatasetParams()
H A Dzip_dataset_op_test.cc28 ZipDatasetParams(std::vector<T> input_dataset_params, in ZipDatasetParams()
H A Dskip_dataset_op_test.cc25 SkipDatasetParams(T input_dataset_params, int64_t count, in SkipDatasetParams()
H A Dfilter_dataset_op_test.cc25 FilterDatasetParams(T input_dataset_params, in FilterDatasetParams()
H A Dprefetch_dataset_op_test.cc28 PrefetchDatasetParams(T input_dataset_params, int64_t buffer_size, in PrefetchDatasetParams()
H A Dflat_map_dataset_op_test.cc25 FlatMapDatasetParams(T input_dataset_params, in FlatMapDatasetParams()
H A Dshard_dataset_op_test.cc25 ShardDatasetParams(T input_dataset_params, int64_t num_shards, int64_t index, in ShardDatasetParams()
H A Dparallel_batch_dataset_op_test.cc29 ParallelBatchDatasetParams(T input_dataset_params, int64_t batch_size, in ParallelBatchDatasetParams()
H A Dfinalize_dataset_op_test.cc28 FinalizeDatasetParams(T input_dataset_params, DataTypeVector output_dtypes, in FinalizeDatasetParams()
H A Drepeat_dataset_op_test.cc33 RepeatDatasetParams(T input_dataset_params, int64_t count, in RepeatDatasetParams()
H A Dcache_dataset_ops_test.cc33 CacheDatasetParams(T input_dataset_params, string filename, in CacheDatasetParams()
H A Dparallel_filter_dataset_op_test.cc32 T input_dataset_params, std::vector<Tensor> other_arguments, in ParallelFilterDatasetParams()
H A Dparallel_map_dataset_op_test.cc28 T input_dataset_params, std::vector<Tensor> other_arguments, in ParallelMapDatasetParams()
H A Dinterleave_dataset_op_test.cc25 InterleaveDatasetParams(T input_dataset_params, in InterleaveDatasetParams()
/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()
H A Dassert_next_dataset_op_test.cc28 AssertNextDatasetParams(T input_dataset_params, in AssertNextDatasetParams()
H A Dunique_dataset_op_test.cc28 UniqueDatasetParams(T input_dataset_params, DataTypeVector output_dtypes, in UniqueDatasetParams()
H A Dassert_prev_dataset_op_test.cc50 AssertPrevDatasetParams(T input_dataset_params, in AssertPrevDatasetParams()
H A Dauto_shard_dataset_op_test.cc32 AutoShardDatasetParams(T input_dataset_params, int64_t num_workers, in AutoShardDatasetParams()
H A Dmap_and_batch_dataset_op_test.cc27 T input_dataset_params, std::vector<Tensor> other_arguments, in MapAndBatchDatasetParams()
H A Dio_ops_test.cc37 SaveDatasetV2Params(T input_dataset_params, const tstring& path, in SaveDatasetV2Params()

12