Home
last modified time | relevance | path

Searched refs:RangeDatasetParams (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Drange_dataset_op_test.cc25 RangeDatasetParams PositiveStepRangeDatasetParams() { in PositiveStepRangeDatasetParams()
26 return RangeDatasetParams(/*start=*/0, /*stop=*/10, /*step=*/3); in PositiveStepRangeDatasetParams()
29 RangeDatasetParams NegativeStepRangeDatasetParams() { in NegativeStepRangeDatasetParams()
30 return RangeDatasetParams(/*start=*/10, /*stop=*/0, /*step=*/-3); in NegativeStepRangeDatasetParams()
33 RangeDatasetParams ZeroStepRangeDatasetParams() { in ZeroStepRangeDatasetParams()
34 return RangeDatasetParams(/*start=*/10, /*stop=*/0, /*step=*/0); in ZeroStepRangeDatasetParams()
37 RangeDatasetParams RangeDatasetParams1() { in RangeDatasetParams1()
38 return RangeDatasetParams(/*start=*/0, /*stop=*/10, /*step=*/3, in RangeDatasetParams1()
42 RangeDatasetParams RangeDatasetParams2() { in RangeDatasetParams2()
43 return RangeDatasetParams(/*start=*/0, /*stop=*/10, /*step=*/3, in RangeDatasetParams2()
[all …]
Dshard_dataset_op_test.cc73 return ShardDatasetParams(RangeDatasetParams(0, 10, 1), in ShardDatasetParams1()
84 return ShardDatasetParams(RangeDatasetParams(0, 10, 1), in ShardDatasetParams2()
95 return ShardDatasetParams(RangeDatasetParams(0, 1, 1), in ShardDatasetParams3()
106 return ShardDatasetParams(RangeDatasetParams(0, 10, 1), in ShardDatasetParams4()
117 return ShardDatasetParams(RangeDatasetParams(0, 10, 1), in ShardDatasetParams5()
129 return ShardDatasetParams(RangeDatasetParams(0, 10, 1), in ShardDatasetParams6()
141 return ShardDatasetParams(RangeDatasetParams(0, 10, 1), in ShardDatasetParams7()
152 return ShardDatasetParams(RangeDatasetParams(0, 10, 1), in InvalidShardDatasetParamsWithNoElemForEachShard()
163 return ShardDatasetParams(RangeDatasetParams(0, 10, 1), in InvalidShardDatasetParams1()
174 return ShardDatasetParams(RangeDatasetParams(0, 10, 1), in InvalidShardDatasetParams2()
[all …]
Dbatch_dataset_op_test.cc27 return BatchDatasetParams(RangeDatasetParams(0, 12, 1), in BatchDatasetParams1()
39 return BatchDatasetParams(RangeDatasetParams(0, 12, 1), in BatchDatasetParams2()
51 return BatchDatasetParams(RangeDatasetParams(0, 10, 1), in BatchDatasetParams3()
63 return BatchDatasetParams(RangeDatasetParams(0, 10, 1), in BatchDatasetParams4()
75 return BatchDatasetParams(RangeDatasetParams(0, 10, 1), in BatchDatasetParams5()
87 return BatchDatasetParams(RangeDatasetParams(0, 10, 1), in BatchDatasetParams6()
99 return BatchDatasetParams(RangeDatasetParams(0, 0, 1), in BatchDatasetParams7()
110 return BatchDatasetParams(RangeDatasetParams(0, 10, 1), in InvalidBatchSizeBatchDatasetParams()
Dparallel_batch_dataset_op_test.cc84 return ParallelBatchDatasetParams(RangeDatasetParams(0, 12, 1), in ParallelBatchDatasetParams1()
96 return ParallelBatchDatasetParams(RangeDatasetParams(0, 12, 1), in ParallelBatchDatasetParams2()
109 RangeDatasetParams(0, 10, 1), in ParallelBatchDatasetParams3()
121 return ParallelBatchDatasetParams(RangeDatasetParams(0, 10, 1), in ParallelBatchDatasetParams4()
134 RangeDatasetParams(0, 10, 1), in ParallelBatchDatasetParams5()
147 RangeDatasetParams(0, 10, 1), in ParallelBatchDatasetParams6()
159 return ParallelBatchDatasetParams(RangeDatasetParams(0, 0, 1), in ParallelBatchDatasetParams7()
170 return ParallelBatchDatasetParams(RangeDatasetParams(0, 12, 1), in ParallelBatchDatasetParams8()
181 return ParallelBatchDatasetParams(RangeDatasetParams(0, 12, 1), in ParallelBatchDatasetParams9()
192 return ParallelBatchDatasetParams(RangeDatasetParams(0, 10, 1), in InvalidBatchSizeParallelBatchDatasetParams()
Dwindow_dataset_op_test.cc80 return WindowDatasetParams(RangeDatasetParams(0, 7, 1), in WindowDatasetParams1()
92 return WindowDatasetParams(RangeDatasetParams(0, 7, 1), in WindowDatasetParams2()
104 return WindowDatasetParams(RangeDatasetParams(0, 7, 1), in WindowDatasetParams3()
116 return WindowDatasetParams(RangeDatasetParams(0, 7, 1), in WindowDatasetParams4()
128 return WindowDatasetParams(RangeDatasetParams(0, 7, 1), in WindowDatasetParams5()
140 return WindowDatasetParams(RangeDatasetParams(0, 7, 1), in WindowDatasetParams6()
152 return WindowDatasetParams(RangeDatasetParams(0, 7, 1), in WindowDatasetParams7()
164 return WindowDatasetParams(RangeDatasetParams(0, 7, 1), in WindowDatasetParams8()
176 return WindowDatasetParams(RangeDatasetParams(0, 7, 1), in WindowDatasetParams9()
188 return WindowDatasetParams(RangeDatasetParams(0, 7, 1), in WindowDatasetParams10()
[all …]
Dshuffle_dataset_op_test.cc101 return ShuffleDatasetParams(RangeDatasetParams(0, 10, 1), in ShuffleDatasetParams1()
114 return ShuffleDatasetParams(RangeDatasetParams(0, 10, 1), in ShuffleDatasetParams2()
128 return ShuffleDatasetParams(RangeDatasetParams(0, 10, 1), in ShuffleDatasetParams3()
141 return ShuffleDatasetParams(RangeDatasetParams(0, 10, 1), in ShuffleDatasetParams4()
155 return ShuffleDatasetParams(RangeDatasetParams(0, 10, 1), in ShuffleDatasetParams5()
168 return ShuffleDatasetParams(RangeDatasetParams(0, 0, 1), in ShuffleDatasetParams6()
182 return ShuffleDatasetParams(RangeDatasetParams(0, 10, 1), in ShuffleDatasetParams7()
196 return ShuffleDatasetParams(RangeDatasetParams(0, 3, 1), in ShuffleDatasetParams8()
208 return ShuffleDatasetParams(RangeDatasetParams(0, 0, 1), in ShuffleDatasetParamsWithInvalidBufferSize()
220 return ShuffleDatasetParams(RangeDatasetParams(0, 0, 1), in ShuffleAndRepeatDatasetParamsWithInvalidBufferSize()
[all …]
Dzip_dataset_op_test.cc75 std::vector<RangeDatasetParams>{RangeDatasetParams(0, 3, 1), in ZipDatasetParams1()
76 RangeDatasetParams(10, 13, 1)}, in ZipDatasetParams1()
86 std::vector<RangeDatasetParams>{RangeDatasetParams(0, 3, 1), in ZipDatasetParams2()
87 RangeDatasetParams(10, 15, 1)}, in ZipDatasetParams2()
Dparallel_map_dataset_op_test.cc112 RangeDatasetParams(0, 10, 3), in ParallelMapDatasetParams1()
130 RangeDatasetParams(0, 10, 3), in ParallelMapDatasetParams2()
148 RangeDatasetParams(0, 10, 3), in ParallelMapDatasetParams3()
166 RangeDatasetParams(0, 10, 3), in ParallelMapDatasetParams4()
184 RangeDatasetParams(0, 10, 3), in ParallelMapDatasetParams5()
202 RangeDatasetParams(0, 10, 3), in ParallelMapDatasetParams6()
221 RangeDatasetParams(0, 10, 3), in ParallelMapDatasetParams7()
241 RangeDatasetParams(0, 10, 3), in ParallelMapDatasetParams8()
257 RangeDatasetParams(0, 10, 3), in ParallelMapDatasetParamsWithInvalidNumParallelCalls()
Dskip_dataset_op_test.cc67 /*input_dataset_params=*/RangeDatasetParams(0, 10, 1), in SkipDatasetParams1()
77 /*input_dataset_params=*/RangeDatasetParams(0, 10, 1), in SkipDatasetParams2()
87 /*input_dataset_params=*/RangeDatasetParams(0, 10, 1), in SkipDatasetParams3()
97 /*input_dataset_params=*/RangeDatasetParams(0, 10, 1), in SkipDatasetParams4()
107 /*input_dataset_params=*/RangeDatasetParams(0, 10, 1), in SkipDatasetParams5()
Dtake_dataset_op_test.cc26 return TakeDatasetParams(RangeDatasetParams(0, 10, 1), in TakeLessTakeDatasetParams()
35 return TakeDatasetParams(RangeDatasetParams(0, 10, 1), in TakeMoreTakeDatasetParams()
44 return TakeDatasetParams(RangeDatasetParams(0, 10, 1), in TakeAllTakeDatasetParams()
53 return TakeDatasetParams(RangeDatasetParams(0, 10, 1), in TakeNothingTakeDatasetParams()
Dmap_dataset_op_test.cc29 RangeDatasetParams(0, 10, 3), in MapDatasetParams1()
56 BatchDatasetParams(RangeDatasetParams(10, 0, -3), in MapDatasetParams2()
81 RangeDatasetParams(0, 10, 3), in MapDatasetParams3()
Dreduce_dataset_op_test.cc99 /*input_dataset_params=*/RangeDatasetParams(0, 10, 1), in ReduceDatasetParams1()
120 /*input_dataset_params=*/RangeDatasetParams(1, 10, 1), in ReduceDatasetParams2()
138 /*input_dataset_params=*/RangeDatasetParams(0, 0, 1), in ReduceDatasetParams3()
Ddataset_test_base.cc963 RangeDatasetParams::RangeDatasetParams( in RangeDatasetParams() function in tensorflow::data::RangeDatasetParams
972 RangeDatasetParams::RangeDatasetParams(int64 start, int64 stop, int64 step) in RangeDatasetParams() function in tensorflow::data::RangeDatasetParams
978 RangeDatasetParams::RangeDatasetParams(int64 start, int64 stop, int64 step, in RangeDatasetParams() function in tensorflow::data::RangeDatasetParams
986 std::vector<Tensor> RangeDatasetParams::GetInputTensors() const { in GetInputTensors()
993 Status RangeDatasetParams::GetInputNames( in GetInputNames()
1000 Status RangeDatasetParams::GetAttributes(AttributeVector* attr_vector) const { in GetAttributes()
1006 string RangeDatasetParams::dataset_type() const { in dataset_type()
Doptimize_dataset_op_test.cc74 TakeDatasetParams(RangeDatasetParams(-3, 3, 1), in TEST_F()
Ddataset_test_base.h177 class RangeDatasetParams : public DatasetParams {
179 RangeDatasetParams(int64 start, int64 stop, int64 step,
184 RangeDatasetParams(int64 start, int64 stop, int64 step);
186 RangeDatasetParams(int64 start, int64 stop, int64 step,
Dpadded_batch_dataset_op_test.cc268 /*input_dataset_params=*/RangeDatasetParams(0, 0, 1), in PaddedBatchDatasetParams7()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Ddirected_interleave_dataset_op_test.cc89 std::vector<RangeDatasetParams>{RangeDatasetParams(0, 3, 1), in AlternateInputsParams()
90 RangeDatasetParams(10, 13, 1)}, in AlternateInputsParams()
104 std::vector<RangeDatasetParams>{RangeDatasetParams(0, 2, 1), in SelectExhaustedInputParams()
105 RangeDatasetParams(10, 13, 1)}, in SelectExhaustedInputParams()
119 std::vector<RangeDatasetParams>{RangeDatasetParams(0, 6, 1)}, in OneInputDatasetParams()
132 /*input_dataset_params_vec=*/std::vector<RangeDatasetParams>{}, in ZeroInputDatasetParams()
148 std::vector<RangeDatasetParams>{RangeDatasetParams(0, 3, 1), in LargeNumInputDatasetsParams()
149 RangeDatasetParams(10, 13, 1)}, in LargeNumInputDatasetsParams()
165 std::vector<RangeDatasetParams>{RangeDatasetParams(0, 3, 1), in SmallNumInputDatasetsParams()
166 RangeDatasetParams(10, 13, 1)}, in SmallNumInputDatasetsParams()
[all …]
Dauto_shard_dataset_op_test.cc82 return AutoShardDatasetParams(RangeDatasetParams(0, 10, 1), in AutoShardDatasetParams1()
94 return AutoShardDatasetParams(RangeDatasetParams(0, 1, 1), in AutoShardDatasetParams2()
107 return AutoShardDatasetParams(RangeDatasetParams(0, 10, 1), in AutoShardDatasetParams3()
122 return AutoShardDatasetParams(RangeDatasetParams(0, 10, 1), in AutoShardDatasetParams4()
134 return AutoShardDatasetParams(RangeDatasetParams(0, 10, 1), in AutoShardDatasetParams5()
146 return AutoShardDatasetParams(RangeDatasetParams(0, 10, 1), in AutoShardDatasetParams6()
158 return AutoShardDatasetParams(RangeDatasetParams(0, 10, 1), in AutoShardDatasetParams7()
Dmap_and_batch_dataset_op_test.cc110 return MapAndBatchDatasetParams(RangeDatasetParams(0, 10, 2), in MapAndBatchDatasetParams1()
127 return MapAndBatchDatasetParams(RangeDatasetParams(0, 10, 2), in MapAndBatchDatasetParams2()
145 RangeDatasetParams(0, 10, 2), in MapAndBatchDatasetParams3()
162 return MapAndBatchDatasetParams(RangeDatasetParams(0, 10, 2), in MapAndBatchDatasetParams4()
180 RangeDatasetParams(0, 10, 2), in MapAndBatchDatasetParams5()
198 RangeDatasetParams(0, 10, 2), in MapAndBatchDatasetParams6()
214 RangeDatasetParams(0, 10, 2), in InvalidNumParallelCallsMapAndBatchDatasetParams()
230 RangeDatasetParams(0, 10, 2), in InvalidBatchSizeMapAndBatchDatasetParams()
Dassert_next_dataset_op_test.cc73 TakeDatasetParams(RangeDatasetParams(0, 10, 1), in AssertNextDatasetParams1()
88 TakeDatasetParams(RangeDatasetParams(0, 10, 1), in AssertNextDatasetParams2()
104 TakeDatasetParams(RangeDatasetParams(0, 10, 1), in InvalidAssertNextDatasetParams()
118 TakeDatasetParams(RangeDatasetParams(0, 10, 1), in ShortAssertNextDatasetParams()
Dsampling_dataset_op_test.cc76 return SamplingDatasetParams(RangeDatasetParams(0, 3, 1), in OneHundredPercentSampleParams()
84 return SamplingDatasetParams(RangeDatasetParams(0, 20, 1), in TenPercentSampleParams()
92 return SamplingDatasetParams(RangeDatasetParams(0, 20, 1), in ZeroPercentSampleParams()