Home
last modified time | relevance | path

Searched defs:input_dataset_params (Results 1 – 25 of 31) sorted by relevance

12

/external/tensorflow/tensorflow/core/data/
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()
/external/tensorflow/tensorflow/core/kernels/data/
Drewrite_dataset_op_test.cc28 RewriteDatasetParams(T input_dataset_params, string rewrite_name, in RewriteDatasetParams()
Doptimize_dataset_op_test.cc28 OptimizeDatasetParams(T input_dataset_params, string optimizations, in OptimizeDatasetParams()
Dget_options_op_test.cc35 GetOptionsParams(T input_dataset_params, DataTypeVector output_dtypes, in GetOptionsParams()
Dreduce_dataset_op_test.cc27 ReduceDatasetParams(T input_dataset_params, std::vector<Tensor> initial_state, in ReduceDatasetParams()
Dzip_dataset_op_test.cc28 ZipDatasetParams(std::vector<T> input_dataset_params, in ZipDatasetParams()
Dskip_dataset_op_test.cc25 SkipDatasetParams(T input_dataset_params, int64_t count, in SkipDatasetParams()
Dfilter_dataset_op_test.cc25 FilterDatasetParams(T input_dataset_params, in FilterDatasetParams()
Dprefetch_dataset_op_test.cc28 PrefetchDatasetParams(T input_dataset_params, int64_t buffer_size, in PrefetchDatasetParams()
Dflat_map_dataset_op_test.cc25 FlatMapDatasetParams(T input_dataset_params, in FlatMapDatasetParams()
Dparallel_batch_dataset_op_test.cc29 ParallelBatchDatasetParams(T input_dataset_params, int64_t batch_size, in ParallelBatchDatasetParams()
Dshard_dataset_op_test.cc25 ShardDatasetParams(T input_dataset_params, int64_t num_shards, int64_t index, in ShardDatasetParams()
Drepeat_dataset_op_test.cc33 RepeatDatasetParams(T input_dataset_params, int64_t count, in RepeatDatasetParams()
Dfinalize_dataset_op_test.cc28 FinalizeDatasetParams(T input_dataset_params, DataTypeVector output_dtypes, in FinalizeDatasetParams()
Dparallel_filter_dataset_op_test.cc32 T input_dataset_params, std::vector<Tensor> other_arguments, in ParallelFilterDatasetParams()
Dcache_dataset_ops_test.cc33 CacheDatasetParams(T input_dataset_params, string filename, in CacheDatasetParams()
Dparallel_map_dataset_op_test.cc28 T input_dataset_params, std::vector<Tensor> other_arguments, in ParallelMapDatasetParams()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dsampling_dataset_op_test.cc28 SamplingDatasetParams(T input_dataset_params, float rate, in SamplingDatasetParams()
Dassert_next_dataset_op_test.cc28 AssertNextDatasetParams(T input_dataset_params, in AssertNextDatasetParams()
Dunique_dataset_op_test.cc28 UniqueDatasetParams(T input_dataset_params, DataTypeVector output_dtypes, in UniqueDatasetParams()
Dassert_prev_dataset_op_test.cc50 AssertPrevDatasetParams(T input_dataset_params, in AssertPrevDatasetParams()
Dauto_shard_dataset_op_test.cc32 AutoShardDatasetParams(T input_dataset_params, int64_t num_workers, in AutoShardDatasetParams()
Dmap_and_batch_dataset_op_test.cc27 T input_dataset_params, std::vector<Tensor> other_arguments, in MapAndBatchDatasetParams()
Dio_ops_test.cc37 SaveDatasetV2Params(T input_dataset_params, const tstring& path, in SaveDatasetV2Params()
Ddirected_interleave_dataset_op_test.cc38 for (auto input_dataset_params : input_dataset_params_vec) { in DirectedInterleaveDatasetParams() local

12