/external/tensorflow/tensorflow/contrib/tpu/python/tpu/ |
H A D | tpu_sharding.py | 137 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 D | tpu_sharding_test.py | 100 _ = 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 D | pyauto_utils.py | 168 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 D | googletest.py | 72 shard_index = int(os.environ['TEST_SHARD_INDEX']) 82 if bucket == shard_index:
|
/external/tensorflow/tensorflow/core/kernels/ |
H A D | topk_op_gpu.cu.cc | 321 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 D | gtest_filter_unittest.py | 215 def RunWithSharding(total_shards, shard_index, command): 218 extra_env = {SHARD_INDEX_ENV_VAR: str(shard_index),
|
H A D | gtest_unittest.cc | 2196 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 D | gtest_filter_unittest.py | 218 def RunWithSharding(total_shards, shard_index, command): 221 extra_env = {SHARD_INDEX_ENV_VAR: str(shard_index),
|
H A D | gtest_unittest.cc | 1896 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 D | gtest_filter_unittest.py | 215 def RunWithSharding(total_shards, shard_index, command): 218 extra_env = {SHARD_INDEX_ENV_VAR: str(shard_index),
|
H A D | gtest_unittest.cc | 2034 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 D | gtest_filter_unittest.py | 215 def RunWithSharding(total_shards, shard_index, command): 218 extra_env = {SHARD_INDEX_ENV_VAR: str(shard_index),
|
H A D | gtest_unittest.cc | 1896 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 D | gtest_filter_unittest.py | 215 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 D | gtest.cc | 2729 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 D | gtest.cc | 3099 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 D | gtest.cc | 3072 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 D | gtest.cc | 2702 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 D | gtest.cc | 2734 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 D | gtest.cc | 2701 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 D | gtest.cc | 3069 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 D | gtest.cc | 2734 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 D | gtest.cc | 4087 // 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 D | gtest-all.cc | 658 // 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 D | gtest-all.cc | 638 // 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...] |