Home
last modified time | relevance | path

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

12

/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 …]
Dparallel_batch_dataset_op_test.cc95 RangeDatasetParams(0, 12, 1), in ParallelBatchDatasetParams1()
110 RangeDatasetParams(0, 12, 1), in ParallelBatchDatasetParams2()
125 RangeDatasetParams(0, 10, 1), in ParallelBatchDatasetParams3()
140 RangeDatasetParams(0, 10, 1), in ParallelBatchDatasetParams4()
155 RangeDatasetParams(0, 10, 1), in ParallelBatchDatasetParams5()
170 RangeDatasetParams(0, 10, 1), in ParallelBatchDatasetParams6()
185 RangeDatasetParams(0, 0, 1), in ParallelBatchDatasetParams7()
199 RangeDatasetParams(0, 12, 1), in ParallelBatchDatasetParams8()
213 RangeDatasetParams(0, 12, 1), in ParallelBatchDatasetParams9()
228 RangeDatasetParams(0, 12, 1), in ParallelBatchDatasetParams10()
[all …]
Dwindow_dataset_op_test.cc85 return WindowDatasetParams(RangeDatasetParams(0, 7, 1), in WindowDatasetParams1()
97 return WindowDatasetParams(RangeDatasetParams(0, 7, 1), in WindowDatasetParams2()
109 return WindowDatasetParams(RangeDatasetParams(0, 7, 1), in WindowDatasetParams3()
121 return WindowDatasetParams(RangeDatasetParams(0, 7, 1), in WindowDatasetParams4()
133 return WindowDatasetParams(RangeDatasetParams(0, 7, 1), in WindowDatasetParams5()
145 return WindowDatasetParams(RangeDatasetParams(0, 7, 1), in WindowDatasetParams6()
157 return WindowDatasetParams(RangeDatasetParams(0, 7, 1), in WindowDatasetParams7()
169 return WindowDatasetParams(RangeDatasetParams(0, 7, 1), in WindowDatasetParams8()
181 return WindowDatasetParams(RangeDatasetParams(0, 7, 1), in WindowDatasetParams9()
193 return WindowDatasetParams(RangeDatasetParams(0, 7, 1), in WindowDatasetParams10()
[all …]
Dshuffle_dataset_op_test.cc105 return ShuffleDatasetParams(RangeDatasetParams(0, 10, 1), in ShuffleDatasetParams1()
118 return ShuffleDatasetParams(RangeDatasetParams(0, 10, 1), in ShuffleDatasetParams2()
132 return ShuffleDatasetParams(RangeDatasetParams(0, 10, 1), in ShuffleDatasetParams3()
145 return ShuffleDatasetParams(RangeDatasetParams(0, 10, 1), in ShuffleDatasetParams4()
159 return ShuffleDatasetParams(RangeDatasetParams(0, 10, 1), in ShuffleDatasetParams5()
172 return ShuffleDatasetParams(RangeDatasetParams(0, 0, 1), in ShuffleDatasetParams6()
186 return ShuffleDatasetParams(RangeDatasetParams(0, 10, 1), in ShuffleDatasetParams7()
200 return ShuffleDatasetParams(RangeDatasetParams(0, 3, 1), in ShuffleDatasetParams8()
212 return ShuffleDatasetParams(RangeDatasetParams(0, 0, 1), in ShuffleDatasetParamsWithInvalidBufferSize()
224 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.cc111 RangeDatasetParams(0, 10, 3), in ParallelMapDatasetParams1()
129 RangeDatasetParams(0, 10, 3), in ParallelMapDatasetParams2()
147 RangeDatasetParams(0, 10, 3), in ParallelMapDatasetParams3()
165 RangeDatasetParams(0, 10, 3), in ParallelMapDatasetParams4()
183 RangeDatasetParams(0, 10, 3), in ParallelMapDatasetParams5()
201 RangeDatasetParams(0, 10, 3), in ParallelMapDatasetParams6()
220 RangeDatasetParams(0, 10, 3), in ParallelMapDatasetParams7()
240 RangeDatasetParams(0, 10, 3), in ParallelMapDatasetParams8()
256 RangeDatasetParams(0, 10, 3), in ParallelMapDatasetParamsWithInvalidNumParallelCalls()
Dbatch_dataset_op_test.cc32 return BatchDatasetParams(RangeDatasetParams(0, 12, 1), in BatchDatasetParams1()
44 return BatchDatasetParams(RangeDatasetParams(0, 12, 1), in BatchDatasetParams2()
56 return BatchDatasetParams(RangeDatasetParams(0, 10, 1), in BatchDatasetParams3()
68 return BatchDatasetParams(RangeDatasetParams(0, 10, 1), in BatchDatasetParams4()
80 return BatchDatasetParams(RangeDatasetParams(0, 10, 1), in BatchDatasetParams5()
92 return BatchDatasetParams(RangeDatasetParams(0, 10, 1), in BatchDatasetParams6()
104 return BatchDatasetParams(RangeDatasetParams(0, 0, 1), in BatchDatasetParams7()
115 return BatchDatasetParams(RangeDatasetParams(0, 10, 1), in InvalidBatchSizeBatchDatasetParams()
Dskip_dataset_op_test.cc68 /*input_dataset_params=*/RangeDatasetParams(0, 10, 1), in SkipDatasetParams1()
78 /*input_dataset_params=*/RangeDatasetParams(0, 10, 1), in SkipDatasetParams2()
88 /*input_dataset_params=*/RangeDatasetParams(0, 10, 1), in SkipDatasetParams3()
98 /*input_dataset_params=*/RangeDatasetParams(0, 10, 1), in SkipDatasetParams4()
108 /*input_dataset_params=*/RangeDatasetParams(0, 10, 1), in SkipDatasetParams5()
Dfinalize_dataset_op_test.cc109 return OptionsDatasetParams(RangeDatasetParams(0, 10, 3), in NoOptimizationOptionsParams()
120 return OptionsDatasetParams(RangeDatasetParams(0, 10, 3), in MaxIntraOpParallelismOptionsParams()
130 return OptionsDatasetParams(RangeDatasetParams(0, 10, 3), in PrivateThreadPoolOptionsParams()
140 return OptionsDatasetParams(RangeDatasetParams(0, 10, 3), in ModelOptionsParams()
150 return OptionsDatasetParams(RangeDatasetParams(0, 10, 3), in OptimizationsDefaultOptionsParams()
160 return OptionsDatasetParams(RangeDatasetParams(0, 10, 3), in AllChainedDatasetsOptionsParams()
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()
Doptions_dataset_op_test.cc35 return OptionsDatasetParams(RangeDatasetParams(0, 10, 3), in OptionsDatasetParams0()
45 return OptionsDatasetParams(RangeDatasetParams(10, 0, -3), in OptionsDatasetParams1()
55 return OptionsDatasetParams(RangeDatasetParams(0, 5, 1), in OptionsDatasetParams2()
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.cc100 /*input_dataset_params=*/RangeDatasetParams(0, 10, 1), in ReduceDatasetParams1()
121 /*input_dataset_params=*/RangeDatasetParams(1, 10, 1), in ReduceDatasetParams2()
139 /*input_dataset_params=*/RangeDatasetParams(0, 0, 1), in ReduceDatasetParams3()
Diterator_ops_test.cc91 RangeDatasetParams dataset_params = RangeDatasetParams(0, 10, 3); in TEST_F()
Drewrite_dataset_op_test.cc68 auto range_dataset_params = RangeDatasetParams(0, 5, 1); in TEST_F()
Doptimize_dataset_op_test.cc74 TakeDatasetParams(RangeDatasetParams(-3, 3, 1), in TEST_F()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Ddirected_interleave_dataset_op_test.cc95 std::vector<RangeDatasetParams>{RangeDatasetParams(0, 3, 1), in AlternateInputsParams()
96 RangeDatasetParams(10, 13, 1)}, in AlternateInputsParams()
112 std::vector<RangeDatasetParams>{RangeDatasetParams(0, 2, 1), in SelectExhaustedInputParams()
113 RangeDatasetParams(10, 13, 1)}, in SelectExhaustedInputParams()
129 std::vector<RangeDatasetParams>{RangeDatasetParams(0, 6, 1)}, in OneInputDatasetParams()
144 /*input_dataset_params_vec=*/std::vector<RangeDatasetParams>{}, in ZeroInputDatasetParams()
160 std::vector<RangeDatasetParams>{RangeDatasetParams(0, 3, 1), in StopOnEmptyDatasetParams()
161 RangeDatasetParams(10, 50, 1)}, in StopOnEmptyDatasetParams()
177 std::vector<RangeDatasetParams>{RangeDatasetParams(0, 3, 1), in SkipEmptyDatasetParams()
178 RangeDatasetParams(10, 50, 1)}, in SkipEmptyDatasetParams()
[all …]
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()
Dauto_shard_dataset_op_test.cc87 return AutoShardDatasetParams(RangeDatasetParams(0, 10, 1), in AutoShardDatasetParams1()
99 return AutoShardDatasetParams(RangeDatasetParams(0, 1, 1), in AutoShardDatasetParams2()
112 return AutoShardDatasetParams(RangeDatasetParams(0, 10, 1), in AutoShardDatasetParams3()
127 return AutoShardDatasetParams(RangeDatasetParams(0, 10, 1), in AutoShardDatasetParams4()
139 return AutoShardDatasetParams(RangeDatasetParams(0, 10, 1), in AutoShardDatasetParams5()
151 return AutoShardDatasetParams(RangeDatasetParams(0, 10, 1), in AutoShardDatasetParams6()
163 return AutoShardDatasetParams(RangeDatasetParams(0, 10, 1), in AutoShardDatasetParams7()
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()
Dassert_prev_dataset_op_test.cc95 TakeDatasetParams(RangeDatasetParams(0, 10, 1), in AssertPrevDatasetParams1()
111 TakeDatasetParams(RangeDatasetParams(0, 10, 1), in AssertPrevDatasetParams2()
150 TakeDatasetParams(RangeDatasetParams(0, 10, 1), in InvalidAssertPrevDatasetParams()
165 TakeDatasetParams(RangeDatasetParams(0, 10, 1), in ShortAssertPrevDatasetParams()
Dsampling_dataset_op_test.cc77 return SamplingDatasetParams(RangeDatasetParams(0, 3, 1), in OneHundredPercentSampleParams()
85 return SamplingDatasetParams(RangeDatasetParams(0, 20, 1), in TenPercentSampleParams()
93 return SamplingDatasetParams(RangeDatasetParams(0, 20, 1), in ZeroPercentSampleParams()
Dio_ops_test.cc116 RangeDatasetParams(0, 10, 2), in SaveDatasetV2Params1()
132 RangeDatasetParams(0, 5, 1), in SaveDatasetV2Params2()
/external/tensorflow/tensorflow/core/data/
Ddataset_test_base.cc995 RangeDatasetParams::RangeDatasetParams( in RangeDatasetParams() function in tensorflow::data::RangeDatasetParams
1004 RangeDatasetParams::RangeDatasetParams(int64_t start, int64_t stop, in RangeDatasetParams() function in tensorflow::data::RangeDatasetParams
1011 RangeDatasetParams::RangeDatasetParams(int64_t start, int64_t stop, in RangeDatasetParams() function in tensorflow::data::RangeDatasetParams
1020 std::vector<Tensor> RangeDatasetParams::GetInputTensors() const { in GetInputTensors()
1027 Status RangeDatasetParams::GetInputNames( in GetInputNames()
1033 Status RangeDatasetParams::GetAttributes(AttributeVector* attr_vector) const { in GetAttributes()
1041 string RangeDatasetParams::dataset_type() const { return "Range"; } in dataset_type()

12