Home
last modified time | relevance | path

Searched defs:shard_index (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/contrib/tpu/python/tpu/
Dtpu_sharding.py137 def get_sharded_shape(self, shape, shard_index=None): argument
/external/autotest/server/cros/ap_configurators/
Dpyauto_utils.py168 def Shard(ilist, shard_index, num_shards): argument
/external/tensorflow/tensorflow/core/kernels/
Dtopk_op_gpu.cu.cc321 int shard_index = max_element.index; in mergeShards() local
333 int shard_index = max_element.index; in mergeShards() local
/external/protobuf/gtest/test/
Dgtest_filter_unittest.py215 def RunWithSharding(total_shards, shard_index, command): argument
Dgtest_unittest.cc2034 for (int shard_index = 0; shard_index < num_shards; shard_index++) { in TEST() local
2048 for (int shard_index = 0; shard_index < num_shards; shard_index++) { in TEST() local
/external/v8/testing/gtest/test/
Dgtest_filter_unittest.py215 def RunWithSharding(total_shards, shard_index, command): argument
Dgtest_unittest.cc1896 for (int shard_index = 0; shard_index < num_shards; shard_index++) { in TEST() local
1910 for (int shard_index = 0; shard_index < num_shards; shard_index++) { in TEST() local
/external/vulkan-validation-layers/tests/gtest-1.7.0/test/
Dgtest_filter_unittest.py215 def RunWithSharding(total_shards, shard_index, command): argument
Dgtest_unittest.cc1889 for (int shard_index = 0; shard_index < num_shards; shard_index++) { in TEST() local
1903 for (int shard_index = 0; shard_index < num_shards; shard_index++) { in TEST() local
/external/google-breakpad/src/testing/gtest/test/
Dgtest_filter_unittest.py215 def RunWithSharding(total_shards, shard_index, command): argument
Dgtest_unittest.cc2196 for (int shard_index = 0; shard_index < num_shards; shard_index++) { in TEST() local
2210 for (int shard_index = 0; shard_index < num_shards; shard_index++) { in TEST() local
/external/googletest/googletest/test/
Dgtest_filter_unittest.py218 def RunWithSharding(total_shards, shard_index, command): argument
Dgtest_unittest.cc1896 for (int shard_index = 0; shard_index < num_shards; shard_index++) { in TEST() local
1910 for (int shard_index = 0; shard_index < num_shards; shard_index++) { in TEST() local
/external/mesa3d/src/gtest/src/
Dgtest.cc2734 const Int32 shard_index = Int32FromEnvOrDie(kTestShardIndex, -1); in OnTestIterationStart() local
4389 const Int32 shard_index = Int32FromEnvOrDie(shard_index_env, -1); in ShouldShard() local
4444 bool ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) { in ShouldRunTestOnShard()
4458 const Int32 shard_index = shard_tests == HONOR_SHARDING_PROTOCOL ? in FilterTests() local
/external/swiftshader/third_party/LLVM/utils/unittest/googletest/
Dgtest.cc2701 const Int32 shard_index = Int32FromEnvOrDie(kTestShardIndex, -1); in OnTestIterationStart() local
4278 const Int32 shard_index = Int32FromEnvOrDie(shard_index_env, -1); in ShouldShard() local
4333 bool ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) { in ShouldRunTestOnShard()
4347 const Int32 shard_index = shard_tests == HONOR_SHARDING_PROTOCOL ? in FilterTests() local
/external/llvm/utils/unittest/googletest/src/
Dgtest.cc2702 const Int32 shard_index = Int32FromEnvOrDie(kTestShardIndex, -1); in OnTestIterationStart() local
4281 const Int32 shard_index = Int32FromEnvOrDie(shard_index_env, -1); in ShouldShard() local
4336 bool ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) { in ShouldRunTestOnShard()
4350 const Int32 shard_index = shard_tests == HONOR_SHARDING_PROTOCOL ? in FilterTests() local
/external/google-breakpad/src/testing/gtest/src/
Dgtest.cc2729 const Int32 shard_index = Int32FromEnvOrDie(kTestShardIndex, -1); in OnTestIterationStart() local
4343 const Int32 shard_index = Int32FromEnvOrDie(shard_index_env, -1); in ShouldShard() local
4398 bool ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) { in ShouldRunTestOnShard()
4412 const Int32 shard_index = shard_tests == HONOR_SHARDING_PROTOCOL ? in FilterTests() local
/external/vulkan-validation-layers/tests/gtest-1.7.0/src/
Dgtest.cc2734 const Int32 shard_index = Int32FromEnvOrDie(kTestShardIndex, -1); in OnTestIterationStart() local
4385 const Int32 shard_index = Int32FromEnvOrDie(shard_index_env, -1); in ShouldShard() local
4440 bool ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) { in ShouldRunTestOnShard()
4454 const Int32 shard_index = shard_tests == HONOR_SHARDING_PROTOCOL ? in FilterTests() local
/external/v8/testing/gtest/src/
Dgtest.cc3069 const Int32 shard_index = Int32FromEnvOrDie(kTestShardIndex, -1); in OnTestIterationStart() local
4721 const Int32 shard_index = Int32FromEnvOrDie(shard_index_env, -1); in ShouldShard() local
4776 bool ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) { in ShouldRunTestOnShard()
4790 const Int32 shard_index = shard_tests == HONOR_SHARDING_PROTOCOL ? in FilterTests() local
/external/googletest/googletest/src/
Dgtest.cc3099 const Int32 shard_index = Int32FromEnvOrDie(kTestShardIndex, -1); in OnTestIterationStart() local
4747 const Int32 shard_index = Int32FromEnvOrDie(shard_index_env, -1); in ShouldShard() local
4802 bool ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) { in ShouldRunTestOnShard()
4816 const Int32 shard_index = shard_tests == HONOR_SHARDING_PROTOCOL ? in FilterTests() local
/external/libvpx/libvpx/third_party/googletest/src/src/
Dgtest.cc3072 const Int32 shard_index = Int32FromEnvOrDie(kTestShardIndex, -1); in OnTestIterationStart() local
4724 const Int32 shard_index = Int32FromEnvOrDie(shard_index_env, -1); in ShouldShard() local
4779 bool ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) { in ShouldRunTestOnShard()
4793 const Int32 shard_index = shard_tests == HONOR_SHARDING_PROTOCOL ? in FilterTests() local
/external/protobuf/gtest/src/
Dgtest.cc4099 const Int32 shard_index = Int32FromEnvOrDie(shard_index_env, -1); in UnitTestImpl() local
4154 bool ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) { in UnitTestImpl()
4168 const Int32 shard_index = shard_tests == HONOR_SHARDING_PROTOCOL ? in UnitTestImpl() local
/external/vulkan-validation-layers/tests/gtest-1.7.0/fused-src/gtest/
Dgtest-all.cc4218 const Int32 shard_index = Int32FromEnvOrDie(kTestShardIndex, -1); in OnTestIterationStart() local
5873 const Int32 shard_index = Int32FromEnvOrDie(shard_index_env, -1); in ShouldShard() local
5928 bool ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) { in ShouldRunTestOnShard()
5942 const Int32 shard_index = shard_tests == HONOR_SHARDING_PROTOCOL ? in FilterTests() local
/external/protobuf/gtest/fused-src/gtest/
Dgtest-all.cc5439 const Int32 shard_index = Int32FromEnvOrDie(shard_index_env, -1); in UnitTestImpl() local
5494 bool ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) { in UnitTestImpl()
5508 const Int32 shard_index = shard_tests == HONOR_SHARDING_PROTOCOL ? in UnitTestImpl() local