Searched refs:shard_index (Results 1 - 25 of 35) sorted by relevance

12

/external/tensorflow/tensorflow/contrib/tpu/python/tpu/
H A Dtpu_sharding.py137 def get_sharded_shape(self, shape, shard_index=None):
146 shard_index: The index of the shard whose shape should be returned.
147 shard_index can be None for sharding policies that use the same
155 ValueError: If shard_index is None when shards are of different
156 shapes; or shard_index is not None and
157 !(0<=shard_index<number_of_shards); or shape does not have at
165 if shard_index is not None:
166 if shard_index < 0 or shard_index >= self.number_of_shards:
167 raise ValueError("shard_index
[all...]
H A Dtpu_sharding_test.py100 _ = p.get_sharded_shape([4, 9], shard_index=4)
102 _ = p.get_sharded_shape([4, 9], shard_index=-1)
108 _ = p.get_sharded_shape([4, 10], shard_index=-1)
/external/autotest/server/cros/ap_configurators/
H A Dpyauto_utils.py168 def Shard(ilist, shard_index, num_shards):
169 """Shard a given list and return the group at index |shard_index|.
173 shard_index: 0-based sharding index
177 chunk_start = shard_index * chunk_size
178 if shard_index == num_shards - 1: # Exhaust the remainder in the last shard.
/external/tensorflow/tensorflow/python/platform/
H A Dgoogletest.py72 shard_index = int(os.environ['TEST_SHARD_INDEX'])
82 if bucket == shard_index:
/external/tensorflow/tensorflow/core/kernels/
H A Dtopk_op_gpu.cu.cc321 int shard_index = max_element.index;
322 top_k_indices[rank] = entries[shard_index].index;
323 int next_shard_index = shard_index + num_shards;
333 int shard_index = max_element.index;
334 top_k_indices[last_k] = entries[shard_index].index;
/external/google-breakpad/src/testing/gtest/test/
H A Dgtest_filter_unittest.py215 def RunWithSharding(total_shards, shard_index, command):
218 extra_env = {SHARD_INDEX_ENV_VAR: str(shard_index),
H A Dgtest_unittest.cc2196 for (int shard_index = 0; shard_index < num_shards; shard_index++) {
2197 if (ShouldRunTestOnShard(num_shards, shard_index, test_id)) {
2199 prev_selected_shard_index = shard_index;
2202 << shard_index << " are both selected to run test " << test_id;
2210 for (int shard_index = 0; shard_index < num_shards; shard_index++) {
2214 ShouldRunTestOnShard(num_shards, shard_index, test_i
[all...]
/external/googletest/googletest/test/
H A Dgtest_filter_unittest.py218 def RunWithSharding(total_shards, shard_index, command):
221 extra_env = {SHARD_INDEX_ENV_VAR: str(shard_index),
H A Dgtest_unittest.cc1896 for (int shard_index = 0; shard_index < num_shards; shard_index++) {
1897 if (ShouldRunTestOnShard(num_shards, shard_index, test_id)) {
1899 prev_selected_shard_index = shard_index;
1902 << shard_index << " are both selected to run test " << test_id;
1910 for (int shard_index = 0; shard_index < num_shards; shard_index++) {
1914 ShouldRunTestOnShard(num_shards, shard_index, test_i
[all...]
/external/protobuf/gtest/test/
H A Dgtest_filter_unittest.py215 def RunWithSharding(total_shards, shard_index, command):
218 extra_env = {SHARD_INDEX_ENV_VAR: str(shard_index),
H A Dgtest_unittest.cc2034 for (int shard_index = 0; shard_index < num_shards; shard_index++) {
2035 if (ShouldRunTestOnShard(num_shards, shard_index, test_id)) {
2037 prev_selected_shard_index = shard_index;
2040 << shard_index << " are both selected to run test " << test_id;
2048 for (int shard_index = 0; shard_index < num_shards; shard_index++) {
2052 ShouldRunTestOnShard(num_shards, shard_index, test_i
[all...]
/external/v8/testing/gtest/test/
H A Dgtest_filter_unittest.py215 def RunWithSharding(total_shards, shard_index, command):
218 extra_env = {SHARD_INDEX_ENV_VAR: str(shard_index),
H A Dgtest_unittest.cc1896 for (int shard_index = 0; shard_index < num_shards; shard_index++) {
1897 if (ShouldRunTestOnShard(num_shards, shard_index, test_id)) {
1899 prev_selected_shard_index = shard_index;
1902 << shard_index << " are both selected to run test " << test_id;
1910 for (int shard_index = 0; shard_index < num_shards; shard_index++) {
1914 ShouldRunTestOnShard(num_shards, shard_index, test_i
[all...]
/external/vulkan-validation-layers/tests/gtest-1.7.0/test/
H A Dgtest_filter_unittest.py215 def RunWithSharding(total_shards, shard_index, command):
218 extra_env = {SHARD_INDEX_ENV_VAR: str(shard_index),
/external/google-breakpad/src/testing/gtest/src/
H A Dgtest.cc2729 const Int32 shard_index = Int32FromEnvOrDie(kTestShardIndex, -1); local
2732 static_cast<int>(shard_index) + 1,
4331 // but inconsistent (i.e., shard_index >= total_shards), prints
4343 const Int32 shard_index = Int32FromEnvOrDie(shard_index_env, -1); local
4345 if (total_shards == -1 && shard_index == -1) {
4347 } else if (total_shards == -1 && shard_index != -1) {
4350 << kTestShardIndex << " = " << shard_index
4355 } else if (total_shards != -1 && shard_index == -1) {
4363 } else if (shard_index < 0 || shard_index >
4398 ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) argument
4412 const Int32 shard_index = shard_tests == HONOR_SHARDING_PROTOCOL ? local
[all...]
/external/googletest/googletest/src/
H A Dgtest.cc3099 const Int32 shard_index = Int32FromEnvOrDie(kTestShardIndex, -1); local
3102 static_cast<int>(shard_index) + 1,
4735 // but inconsistent (i.e., shard_index >= total_shards), prints
4747 const Int32 shard_index = Int32FromEnvOrDie(shard_index_env, -1); local
4749 if (total_shards == -1 && shard_index == -1) {
4751 } else if (total_shards == -1 && shard_index != -1) {
4754 << kTestShardIndex << " = " << shard_index
4759 } else if (total_shards != -1 && shard_index == -1) {
4767 } else if (shard_index < 0 || shard_index >
4802 ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) argument
4816 const Int32 shard_index = shard_tests == HONOR_SHARDING_PROTOCOL ? local
[all...]
/external/libvpx/libvpx/third_party/googletest/src/src/
H A Dgtest.cc3072 const Int32 shard_index = Int32FromEnvOrDie(kTestShardIndex, -1); local
3075 static_cast<int>(shard_index) + 1,
4712 // but inconsistent (i.e., shard_index >= total_shards), prints
4724 const Int32 shard_index = Int32FromEnvOrDie(shard_index_env, -1); local
4726 if (total_shards == -1 && shard_index == -1) {
4728 } else if (total_shards == -1 && shard_index != -1) {
4731 << kTestShardIndex << " = " << shard_index
4736 } else if (total_shards != -1 && shard_index == -1) {
4744 } else if (shard_index < 0 || shard_index >
4779 ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) argument
4793 const Int32 shard_index = shard_tests == HONOR_SHARDING_PROTOCOL ? local
[all...]
/external/llvm/utils/unittest/googletest/src/
H A Dgtest.cc2702 const Int32 shard_index = Int32FromEnvOrDie(kTestShardIndex, -1); local
2705 static_cast<int>(shard_index) + 1,
4269 // but inconsistent (i.e., shard_index >= total_shards), prints
4281 const Int32 shard_index = Int32FromEnvOrDie(shard_index_env, -1); local
4283 if (total_shards == -1 && shard_index == -1) {
4285 } else if (total_shards == -1 && shard_index != -1) {
4288 << kTestShardIndex << " = " << shard_index
4293 } else if (total_shards != -1 && shard_index == -1) {
4301 } else if (shard_index < 0 || shard_index >
4336 ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) argument
4350 const Int32 shard_index = shard_tests == HONOR_SHARDING_PROTOCOL ? local
[all...]
/external/mesa3d/src/gtest/src/
H A Dgtest.cc2734 const Int32 shard_index = Int32FromEnvOrDie(kTestShardIndex, -1); local
2737 static_cast<int>(shard_index) + 1,
4377 // but inconsistent (i.e., shard_index >= total_shards), prints
4389 const Int32 shard_index = Int32FromEnvOrDie(shard_index_env, -1); local
4391 if (total_shards == -1 && shard_index == -1) {
4393 } else if (total_shards == -1 && shard_index != -1) {
4396 << kTestShardIndex << " = " << shard_index
4401 } else if (total_shards != -1 && shard_index == -1) {
4409 } else if (shard_index < 0 || shard_index >
4444 ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) argument
4458 const Int32 shard_index = shard_tests == HONOR_SHARDING_PROTOCOL ? local
[all...]
/external/swiftshader/third_party/LLVM/utils/unittest/googletest/
H A Dgtest.cc2701 const Int32 shard_index = Int32FromEnvOrDie(kTestShardIndex, -1); local
2704 static_cast<int>(shard_index) + 1,
4266 // but inconsistent (i.e., shard_index >= total_shards), prints
4278 const Int32 shard_index = Int32FromEnvOrDie(shard_index_env, -1); local
4280 if (total_shards == -1 && shard_index == -1) {
4282 } else if (total_shards == -1 && shard_index != -1) {
4285 << kTestShardIndex << " = " << shard_index
4290 } else if (total_shards != -1 && shard_index == -1) {
4298 } else if (shard_index < 0 || shard_index >
4333 ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) argument
4347 const Int32 shard_index = shard_tests == HONOR_SHARDING_PROTOCOL ? local
[all...]
/external/v8/testing/gtest/src/
H A Dgtest.cc3069 const Int32 shard_index = Int32FromEnvOrDie(kTestShardIndex, -1); local
3072 static_cast<int>(shard_index) + 1,
4709 // but inconsistent (i.e., shard_index >= total_shards), prints
4721 const Int32 shard_index = Int32FromEnvOrDie(shard_index_env, -1); local
4723 if (total_shards == -1 && shard_index == -1) {
4725 } else if (total_shards == -1 && shard_index != -1) {
4728 << kTestShardIndex << " = " << shard_index
4733 } else if (total_shards != -1 && shard_index == -1) {
4741 } else if (shard_index < 0 || shard_index >
4776 ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) argument
4790 const Int32 shard_index = shard_tests == HONOR_SHARDING_PROTOCOL ? local
[all...]
/external/vulkan-validation-layers/tests/gtest-1.7.0/src/
H A Dgtest.cc2734 const Int32 shard_index = Int32FromEnvOrDie(kTestShardIndex, -1); local
2737 static_cast<int>(shard_index) + 1,
4373 // but inconsistent (i.e., shard_index >= total_shards), prints
4385 const Int32 shard_index = Int32FromEnvOrDie(shard_index_env, -1); local
4387 if (total_shards == -1 && shard_index == -1) {
4389 } else if (total_shards == -1 && shard_index != -1) {
4392 << kTestShardIndex << " = " << shard_index
4397 } else if (total_shards != -1 && shard_index == -1) {
4405 } else if (shard_index < 0 || shard_index >
4440 ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) argument
4454 const Int32 shard_index = shard_tests == HONOR_SHARDING_PROTOCOL ? local
[all...]
/external/protobuf/gtest/src/
H A Dgtest.cc4087 // but inconsistent (i.e., shard_index >= total_shards), prints
4099 const Int32 shard_index = Int32FromEnvOrDie(shard_index_env, -1); local
4101 if (total_shards == -1 && shard_index == -1) {
4103 } else if (total_shards == -1 && shard_index != -1) {
4106 << kTestShardIndex << " = " << shard_index
4111 } else if (total_shards != -1 && shard_index == -1) {
4119 } else if (shard_index < 0 || shard_index >= total_shards) {
4123 << ", but you have " << kTestShardIndex << "=" << shard_index
4153 // method. Assumes that 0 <= shard_index < total_shard
4154 ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) argument
4168 const Int32 shard_index = shard_tests == HONOR_SHARDING_PROTOCOL ? local
[all...]
/external/vulkan-validation-layers/tests/gtest-1.7.0/fused-src/gtest/
H A Dgtest-all.cc658 // but inconsistent (e.g., shard_index >= total_shards), prints
674 // method. Assumes that 0 <= shard_index < total_shards.
676 int total_shards, int shard_index, int test_id);
4218 const Int32 shard_index = Int32FromEnvOrDie(kTestShardIndex, -1); local
4221 static_cast<int>(shard_index) + 1,
5861 // but inconsistent (i.e., shard_index >= total_shards), prints
5873 const Int32 shard_index = Int32FromEnvOrDie(shard_index_env, -1); local
5875 if (total_shards == -1 && shard_index == -1) {
5877 } else if (total_shards == -1 && shard_index != -1) {
5880 << kTestShardIndex << " = " << shard_index
5928 ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) argument
5942 const Int32 shard_index = shard_tests == HONOR_SHARDING_PROTOCOL ? local
[all...]
/external/protobuf/gtest/fused-src/gtest/
H A Dgtest-all.cc638 // but inconsistent (e.g., shard_index >= total_shards), prints
654 // method. Assumes that 0 <= shard_index < total_shards.
656 int total_shards, int shard_index, int test_id);
5427 // but inconsistent (i.e., shard_index >= total_shards), prints
5439 const Int32 shard_index = Int32FromEnvOrDie(shard_index_env, -1); local
5441 if (total_shards == -1 && shard_index == -1) {
5443 } else if (total_shards == -1 && shard_index != -1) {
5446 << kTestShardIndex << " = " << shard_index
5451 } else if (total_shards != -1 && shard_index == -1) {
5459 } else if (shard_index <
5494 ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) argument
5508 const Int32 shard_index = shard_tests == HONOR_SHARDING_PROTOCOL ? local
[all...]

Completed in 651 milliseconds

12