| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | dataset_test_base.h | 121 class DatasetParams {
123 DatasetParams(DataTypeVector output_dtypes,
127 virtual ~DatasetParams() {} in ~DatasetParams()
151 const std::vector<std::shared_ptr<DatasetParams>>& input_dataset_params() in input_dataset_params()
167 std::vector<std::shared_ptr<DatasetParams>> input_dataset_params_;
177 class RangeDatasetParams : public DatasetParams {
205 class BatchDatasetParams : public DatasetParams {
213 : DatasetParams(std::move(output_dtypes), std::move(output_shapes), in BatchDatasetParams()
241 class MapDatasetParams : public DatasetParams {
251 : DatasetParams(std::move(output_dtypes), std::move(output_shapes), in MapDatasetParams()
[all …]
|
| D | dataset_test_base.cc | 632 const DatasetParams& params, const std::vector<Tensor>& expected_outputs) { in CheckSplitProviderFullIteration()
646 const DatasetParams& params, int64 num_shards, int64 shard_index, in CheckSplitProviderShardedIteration()
767 Status DatasetOpsTestBase::Initialize(const DatasetParams& dataset_params) { in Initialize()
785 const DatasetParams& dataset_params) { in InitializeRuntime()
792 Status DatasetOpsTestBase::MakeDataset(const DatasetParams& dataset_params, in MakeDataset()
809 const DatasetParams& dataset_params, in RunDatasetOp()
842 const DatasetParams& dataset_params, in MakeDataset()
858 const DatasetParams& dataset_params, const TestDataset& dataset, in MakeIterator()
877 const DatasetParams& dataset_params, const TestDataset& dataset, in MakeIterator()
883 Status DatasetOpsTestBase::RunDatasetOp(const DatasetParams& dataset_params, in RunDatasetOp()
[all …]
|
| D | optimize_dataset_op_test.cc | 25 class OptimizeDatasetParams : public DatasetParams {
33 : DatasetParams(std::move(output_dtypes), std::move(output_shapes), in OptimizeDatasetParams()
|
| D | reduce_dataset_op_test.cc | 24 class ReduceDatasetParams : public DatasetParams {
35 : DatasetParams(std::move(output_dtypes), std::move(output_shapes), in ReduceDatasetParams()
|
| D | skip_dataset_op_test.cc | 22 class SkipDatasetParams : public DatasetParams {
29 : DatasetParams(std::move(output_dtypes), std::move(output_shapes), in SkipDatasetParams()
|
| D | zip_dataset_op_test.cc | 25 class ZipDatasetParams : public DatasetParams {
32 : DatasetParams(std::move(output_dtypes), std::move(output_shapes), in ZipDatasetParams()
|
| D | cache_dataset_ops_test.cc | 26 class CacheDatasetParams : public DatasetParams {
33 : DatasetParams(std::move(output_dtypes), std::move(output_shapes), in CacheDatasetParams()
69 Status Initialize(const DatasetParams& dataset_params) { in Initialize()
|
| D | flat_map_dataset_op_test.cc | 22 class FlatMapDatasetParams : public DatasetParams {
33 : DatasetParams(std::move(output_dtypes), std::move(output_shapes), in FlatMapDatasetParams()
|
| D | prefetch_dataset_op_test.cc | 25 class PrefetchDatasetParams : public DatasetParams {
33 : DatasetParams(std::move(output_dtypes), std::move(output_shapes), in PrefetchDatasetParams()
|
| D | text_line_dataset_op_test.cc | 28 class TextLineDatasetParams : public DatasetParams {
33 : DatasetParams({DT_STRING}, {PartialTensorShape({})}, in TextLineDatasetParams()
|
| D | filter_dataset_op_test.cc | 22 class FilterDatasetParams : public DatasetParams {
33 : DatasetParams(std::move(output_dtypes), std::move(output_shapes), in FilterDatasetParams()
|
| D | tf_record_dataset_op_test.cc | 22 class TFRecordDatasetParams : public DatasetParams {
27 : DatasetParams({DT_STRING}, {PartialTensorShape({})}, in TFRecordDatasetParams()
|
| D | shard_dataset_op_test.cc | 22 class ShardDatasetParams : public DatasetParams {
29 : DatasetParams(std::move(output_dtypes), std::move(output_shapes), in ShardDatasetParams()
|
| D | fixed_length_record_dataset_op_test.cc | 29 class FixedLengthRecordDatasetParams : public DatasetParams {
36 : DatasetParams({DT_STRING}, {PartialTensorShape({})}, in FixedLengthRecordDatasetParams()
|
| D | parallel_batch_dataset_op_test.cc | 26 class ParallelBatchDatasetParams : public DatasetParams {
34 : DatasetParams(std::move(output_dtypes), std::move(output_shapes), in ParallelBatchDatasetParams()
|
| D | map_defun_op_test.cc | 23 class MapDefunOpParams : public DatasetParams {
33 : DatasetParams(std::move(output_dtypes), std::move(output_shapes), in MapDefunOpParams()
|
| D | tensor_dataset_op_test.cc | 26 class TensorDatasetParams : public DatasetParams {
29 : DatasetParams(TensorDtypes(components), TensorShapes(components), in TensorDatasetParams()
|
| D | repeat_dataset_op_test.cc | 26 class RepeatDatasetParams : public DatasetParams {
33 : DatasetParams(std::move(output_dtypes), std::move(output_shapes), in RepeatDatasetParams()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | auto_shard_dataset_op_test.cc | 24 class AutoShardDatasetParams : public DatasetParams {
32 : DatasetParams(std::move(output_dtypes), std::move(output_shapes), in AutoShardDatasetParams()
|
| D | sampling_dataset_op_test.cc | 25 class SamplingDatasetParams : public DatasetParams {
32 : DatasetParams(std::move(output_dtypes), std::move(output_shapes), in SamplingDatasetParams()
|
| D | assert_next_dataset_op_test.cc | 25 class AssertNextDatasetParams : public DatasetParams {
33 : DatasetParams(std::move(output_dtypes), std::move(output_shapes), in AssertNextDatasetParams()
|
| D | random_dataset_op_test.cc | 45 class RandomDatasetParams : public DatasetParams {
50 : DatasetParams(std::move(output_dtypes), std::move(output_shapes), in RandomDatasetParams()
|
| D | unique_dataset_op_test.cc | 25 class UniqueDatasetParams : public DatasetParams {
30 : DatasetParams(std::move(output_dtypes), std::move(output_shapes), in UniqueDatasetParams()
|
| D | lmdb_dataset_op_test.cc | 33 class LMDBDatasetParams : public DatasetParams {
39 : DatasetParams(std::move(output_dtypes), std::move(output_shapes), in LMDBDatasetParams()
|
| D | directed_interleave_dataset_op_test.cc | 23 class DirectedInterleaveDatasetParams : public DatasetParams {
31 : DatasetParams(std::move(output_dtypes), std::move(output_shapes), in DirectedInterleaveDatasetParams()
|