Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/data/
Doptimize_dataset_op_test.cc28 OptimizeDatasetParams(T input_dataset_params, string optimizations, in OptimizeDatasetParams()
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()
244 MapDatasetParams(T input_dataset_params, std::vector<Tensor> other_arguments, in MapDatasetParams()
317 TakeDatasetParams(T input_dataset_params, int count, in TakeDatasetParams()
Dreduce_dataset_op_test.cc27 ReduceDatasetParams(T input_dataset_params, std::vector<Tensor> initial_state, in ReduceDatasetParams()
Dskip_dataset_op_test.cc25 SkipDatasetParams(T input_dataset_params, int64 count, in SkipDatasetParams()
Dzip_dataset_op_test.cc28 ZipDatasetParams(std::vector<T> input_dataset_params, in ZipDatasetParams()
Dflat_map_dataset_op_test.cc25 FlatMapDatasetParams(T input_dataset_params, in FlatMapDatasetParams()
Dfilter_dataset_op_test.cc25 FilterDatasetParams(T input_dataset_params, in FilterDatasetParams()
Dparallel_batch_dataset_op_test.cc29 ParallelBatchDatasetParams(T input_dataset_params, int64 batch_size, in ParallelBatchDatasetParams()
Dprefetch_dataset_op_test.cc28 PrefetchDatasetParams(T input_dataset_params, int64 buffer_size, in PrefetchDatasetParams()
Dshard_dataset_op_test.cc25 ShardDatasetParams(T input_dataset_params, int64 num_shards, int64 index, in ShardDatasetParams()
Drepeat_dataset_op_test.cc29 RepeatDatasetParams(T input_dataset_params, int64 count, in RepeatDatasetParams()
Dinterleave_dataset_op_test.cc25 InterleaveDatasetParams(T input_dataset_params, in InterleaveDatasetParams()
Dcache_dataset_ops_test.cc29 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()
Dparallel_interleave_dataset_op_test.cc27 T input_dataset_params, std::vector<Tensor> other_arguments, in ParallelInterleaveDatasetParams()
Dshuffle_dataset_op_test.cc27 ShuffleDatasetParams(T input_dataset_params, int64 buffer_size, int64 seed, in ShuffleDatasetParams()
Dwindow_dataset_op_test.cc26 WindowDatasetParams(T input_dataset_params, int64 size, int64 shift, in WindowDatasetParams()
Dpadded_batch_dataset_op_test.cc28 PaddedBatchDatasetParams(T input_dataset_params, int64 batch_size, in PaddedBatchDatasetParams()
/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()
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()
Dmap_and_batch_dataset_op_test.cc27 T input_dataset_params, std::vector<Tensor> other_arguments, in MapAndBatchDatasetParams()
Ddirected_interleave_dataset_op_test.cc36 for (auto input_dataset_params : input_dataset_params_vec) { in DirectedInterleaveDatasetParams() local
Dparallel_interleave_dataset_op_test.cc29 T input_dataset_params, std::vector<Tensor> other_arguments, in ParallelInterleaveDatasetParams()