Home
last modified time | relevance | path

Searched refs:shard_index (Results 1 – 13 of 13) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/
Dconcat_op.cc63 #define f(fv, sv, shard_index) \ argument
64 (((fv) == -1 && (sv) == -1) || ((fv) < (sv) && (shard_index) >= (fv) && (shard_index) < (sv)) || \
65 ((fv) > (sv) && ((shard_index) >= (fv) || (shard_index) < (sv)))) // NOLINT
/third_party/skia/third_party/externals/angle2/src/tests/restricted_traces/
Drestricted_trace_gold_tests.py413 def _shard_tests(tests, shard_count, shard_index): argument
414 return [tests[index] for index in range(shard_index, len(tests), shard_count)]
458 args.shard_index = int(env.pop('GTEST_SHARD_INDEX'))
490 sharded_tests = _shard_tests(tests['traces'], args.shard_count, args.shard_index)
/third_party/skia/third_party/externals/angle2/src/tests/
Drun_perf_tests.py108 def _shard_tests(tests, shard_count, shard_index): argument
109 return [tests[index] for index in range(shard_index, len(tests), shard_count)]
320 args.shard_index = int(env.pop('GTEST_SHARD_INDEX'))
333 tests = _shard_tests(tests, args.shard_count, args.shard_index)
/third_party/mindspore/mindspore/ccsrc/minddata/mindrecord/meta/
Dshard_header.cc43 int shard_index = 0; in InitializeHeader() local
56 RETURN_IF_NOT_OK(ParsePage(header["page"], shard_index, load_dataset)); in InitializeHeader()
57 shard_index++; in InitializeHeader()
226 Status ShardHeader::ParsePage(const json &pages, int shard_index, bool load_dataset) { in ParsePage() argument
254 pages_[shard_index].push_back(std::move(parsed_page)); in ParsePage()
/third_party/googletest/googletest/test/
Dgoogletest-filter-unittest.py220 def RunWithSharding(total_shards, shard_index, command): argument
223 extra_env = {SHARD_INDEX_ENV_VAR: str(shard_index),
Dgtest_unittest.cc1961 for (int shard_index = 0; shard_index < num_shards; shard_index++) { in TEST() local
1962 if (ShouldRunTestOnShard(num_shards, shard_index, test_id)) { in TEST()
1964 prev_selected_shard_index = shard_index; in TEST()
1967 << shard_index << " are both selected to run test " << test_id; in TEST()
1975 for (int shard_index = 0; shard_index < num_shards; shard_index++) { in TEST() local
1979 ShouldRunTestOnShard(num_shards, shard_index, test_id); in TEST()
/third_party/mesa3d/src/gtest/src/
Dgtest.cc3156 const Int32 shard_index = Int32FromEnvOrDie(kTestShardIndex, -1); in OnTestIterationStart() local
3159 static_cast<int>(shard_index) + 1, in OnTestIterationStart()
5426 const Int32 shard_index = Int32FromEnvOrDie(shard_index_env, -1); in ShouldShard() local
5428 if (total_shards == -1 && shard_index == -1) { in ShouldShard()
5430 } else if (total_shards == -1 && shard_index != -1) { in ShouldShard()
5433 << kTestShardIndex << " = " << shard_index in ShouldShard()
5438 } else if (total_shards != -1 && shard_index == -1) { in ShouldShard()
5446 } else if (shard_index < 0 || shard_index >= total_shards) { in ShouldShard()
5450 << ", but you have " << kTestShardIndex << "=" << shard_index in ShouldShard()
5481 bool ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) { in ShouldRunTestOnShard() argument
[all …]
Dgtest-internal-inl.h273 int total_shards, int shard_index, int test_id);
/third_party/mindspore/mindspore/ccsrc/minddata/mindrecord/include/
Dshard_header.h164 Status ParsePage(const json &page, int shard_index, bool load_dataset);
/third_party/googletest/googletest/src/
Dgtest.cc3403 const int32_t shard_index = Int32FromEnvOrDie(kTestShardIndex, -1); in OnTestIterationStart() local
3405 static_cast<int>(shard_index) + 1, in OnTestIterationStart()
5963 const int32_t shard_index = Int32FromEnvOrDie(shard_index_env, -1); in ShouldShard() local
5965 if (total_shards == -1 && shard_index == -1) { in ShouldShard()
5967 } else if (total_shards == -1 && shard_index != -1) { in ShouldShard()
5970 << kTestShardIndex << " = " << shard_index in ShouldShard()
5975 } else if (total_shards != -1 && shard_index == -1) { in ShouldShard()
5983 } else if (shard_index < 0 || shard_index >= total_shards) { in ShouldShard()
5987 << ", but you have " << kTestShardIndex << "=" << shard_index in ShouldShard()
6018 bool ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) { in ShouldRunTestOnShard() argument
[all …]
Dgtest-internal-inl.h282 int total_shards, int shard_index, int test_id);
/third_party/cef/tools/distrib/gtest/
Dgtest-all.cc670 int total_shards, int shard_index, int test_id);
4894 const int32_t shard_index = Int32FromEnvOrDie(kTestShardIndex, -1); in OnTestIterationStart() local
4896 static_cast<int>(shard_index) + 1, in OnTestIterationStart()
7449 const int32_t shard_index = Int32FromEnvOrDie(shard_index_env, -1); in ShouldShard() local
7451 if (total_shards == -1 && shard_index == -1) { in ShouldShard()
7453 } else if (total_shards == -1 && shard_index != -1) { in ShouldShard()
7456 << kTestShardIndex << " = " << shard_index in ShouldShard()
7461 } else if (total_shards != -1 && shard_index == -1) { in ShouldShard()
7469 } else if (shard_index < 0 || shard_index >= total_shards) { in ShouldShard()
7473 << ", but you have " << kTestShardIndex << "=" << shard_index in ShouldShard()
[all …]
/third_party/mindspore/mindspore/lite/tools/dataset/cropper/
Ddependencies.txt1 …tasetops/rename_node.cc", "mindspore/ccsrc/minddata/mindrecord/include/shard_index.h", "mindspore/…