| /third_party/mindspore/tests/ut/python/dataset/ |
| D | test_sampler.py | 31 def test_config(num_samples, num_repeats=None): argument
34 if num_repeats is not None:
35 data1 = data1.repeat(num_repeats)
45 assert test_config(num_samples=3, num_repeats=None) == [0, 1, 2]
46 assert test_config(num_samples=None, num_repeats=2) == [0, 1, 2, 3, 4] * 2
47 assert test_config(num_samples=4, num_repeats=2) == [0, 1, 2, 3] * 2
54 def test_config(replacement, num_samples, num_repeats): argument
57 data1 = data1.repeat(num_repeats)
66 assert len(set(test_config(replacement=False, num_samples=2, num_repeats=6))) > 2
69 assert sorted(test_config(replacement=False, num_samples=None, num_repeats=4)) == ordered_res
[all …]
|
| D | test_datasets_tfrecord.py | 185 def get_res(shard_id, num_repeats): argument
188 data1 = data1.repeat(num_repeats)
214 def get_res(num_shards, shard_id, num_repeats): argument
216 ds1 = ds1.repeat(num_repeats)
|
| /third_party/mindspore/tests/ut/cpp/dataset/ |
| D | map_op_test.cc | 389 uint32_t num_repeats = 3; in TEST_F() local
400 std::shared_ptr<RepeatOp> my_repeat_op = std::make_shared<RepeatOp>(num_repeats); in TEST_F()
418 my_tfreader_op->SetTotalRepeats(num_repeats); in TEST_F()
419 my_tfreader_op->SetNumRepeatsPerEpoch(num_repeats); in TEST_F()
444 ASSERT_EQ(row_count, 10 * num_repeats); in TEST_F()
450 uint32_t num_repeats = 3; in TEST_F() local
461 std::shared_ptr<RepeatOp> my_repeat_op = std::make_shared<RepeatOp>(num_repeats); in TEST_F()
475 my_map_op->SetTotalRepeats(num_repeats); in TEST_F()
476 my_map_op->SetNumRepeatsPerEpoch(num_repeats); in TEST_F()
480 my_tfreader_op->SetTotalRepeats(num_repeats); in TEST_F()
[all …]
|
| D | c_api_epoch_ctrl_test.cc | 126 int32_t num_repeats = 3; in TEST_F() local
134 ds = ds->Repeat(num_repeats); in TEST_F()
161 EXPECT_EQ(i, sampler_size * num_repeats * num_epochs); in TEST_F()
175 int32_t num_repeats[2] = {2, 3}; in TEST_F() local
182 ds = ds->Repeat(num_repeats[0]); in TEST_F()
183 ds = ds->Repeat(num_repeats[1]); in TEST_F()
210 EXPECT_EQ(i, sampler_size * num_repeats[0] * num_repeats[1] * num_epochs); in TEST_F()
|
| D | cache_op_test.cc | 282 uint32_t num_repeats = 4; in TEST_F() local
283 std::shared_ptr<RepeatOp> myRepeatOp = std::make_shared<RepeatOp>(num_repeats); in TEST_F()
288 myCacheOp->SetTotalRepeats(num_repeats); in TEST_F()
289 myCacheOp->SetNumRepeatsPerEpoch(num_repeats); in TEST_F()
395 uint32_t num_repeats = 4; in TEST_F() local
396 std::shared_ptr<RepeatOp> myRepeatOp = std::make_shared<RepeatOp>(num_repeats); in TEST_F()
401 myCacheOp->SetTotalRepeats(num_repeats); in TEST_F()
402 myCacheOp->SetNumRepeatsPerEpoch(num_repeats); in TEST_F()
484 uint32_t num_repeats = 4; in TEST_F() local
485 std::shared_ptr<RepeatOp> myRepeatOp = std::make_shared<RepeatOp>(num_repeats); in TEST_F()
[all …]
|
| D | zip_op_test.cc | 135 uint32_t num_repeats = 3; in TEST_F() local
165 my_tfreader_op->SetTotalRepeats(num_repeats); in TEST_F()
166 my_tfreader_op->SetNumRepeatsPerEpoch(num_repeats); in TEST_F()
169 my_tfreader_op2->SetTotalRepeats(num_repeats); in TEST_F()
170 my_tfreader_op2->SetNumRepeatsPerEpoch(num_repeats); in TEST_F()
174 std::shared_ptr<RepeatOp> my_repeat_op = std::make_shared<RepeatOp>(num_repeats); in TEST_F()
177 zip_op->SetTotalRepeats(num_repeats); in TEST_F()
178 zip_op->SetNumRepeatsPerEpoch(num_repeats); in TEST_F()
|
| D | mind_record_op_test.cc | 354 uint32_t num_repeats = 2; in TEST_F() local
355 std::shared_ptr<RepeatOp> my_repeat_op = std::make_shared<RepeatOp>(num_repeats); in TEST_F()
359 my_mindrecord_op->SetTotalRepeats(num_repeats); in TEST_F()
360 my_mindrecord_op->SetNumRepeatsPerEpoch(num_repeats); in TEST_F()
427 uint32_t num_repeats = 2; in TEST_F() local
428 std::shared_ptr<RepeatOp> my_repeat_op = std::make_shared<RepeatOp>(num_repeats); in TEST_F()
432 my_mindrecord_op->SetTotalRepeats(num_repeats); in TEST_F()
433 my_mindrecord_op->SetNumRepeatsPerEpoch(num_repeats); in TEST_F()
|
| D | shuffle_op_test.cc | 283 uint32_t num_repeats = 2; in TEST_F() local
284 std::shared_ptr<RepeatOp> my_repeat_op = std::make_shared<RepeatOp>(num_repeats); in TEST_F()
289 my_shuffle_op->SetTotalRepeats(num_repeats); in TEST_F()
290 my_shuffle_op->SetNumRepeatsPerEpoch(num_repeats); in TEST_F()
293 my_tfreader_op->SetTotalRepeats(num_repeats); in TEST_F()
294 my_tfreader_op->SetNumRepeatsPerEpoch(num_repeats); in TEST_F()
|
| D | tfReader_op_test.cc | 347 uint32_t num_repeats = 3; in TEST_F() local
348 std::shared_ptr<RepeatOp> my_repeat_op = std::make_shared<RepeatOp>(num_repeats); in TEST_F()
353 my_tfreader_op->SetTotalRepeats(num_repeats); in TEST_F()
354 my_tfreader_op->SetNumRepeatsPerEpoch(num_repeats); in TEST_F()
|
| /third_party/openssl/test/ |
| D | ecstresstest.c | 21 static int64_t num_repeats; variable
80 || !TEST_ptr(result = walk_curve(group, point, num_repeats))) in test_curve()
129 if (!atoi64(NUM_REPEATS, &num_repeats)) { in setup_tests()
139 if (!atoi64(p, &num_repeats) in setup_tests()
140 || num_repeats < 0) in setup_tests()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/ |
| D | repeat_op.h | 94 int32_t num_repeats() { return num_repeats_; } in num_repeats() function
|