| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_sharding.py | 137 def get_sharded_shape(self, shape, shard_index=None): argument
165 if shard_index is not None:
166 if shard_index < 0 or shard_index >= self.number_of_shards:
168 (shard_index, self._number_of_shards))
|
| 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/ |
| D | pyauto_utils.py | 168 def Shard(ilist, shard_index, num_shards): argument
177 chunk_start = shard_index * chunk_size
178 if shard_index == num_shards - 1: # Exhaust the remainder in the last shard.
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | topk_op_gpu.h | 323 int shard_index = max_element.index;
324 top_k_indices[rank] = entries[shard_index].index;
325 int next_shard_index = shard_index + num_shards;
335 int shard_index = max_element.index;
336 top_k_indices[last_k] = entries[shard_index].index;
|
| /external/swiftshader/third_party/LLVM/utils/unittest/googletest/ |
| D | gtest.cc | 2701 const Int32 shard_index = Int32FromEnvOrDie(kTestShardIndex, -1); in OnTestIterationStart() local
2704 static_cast<int>(shard_index) + 1, in OnTestIterationStart()
4278 const Int32 shard_index = Int32FromEnvOrDie(shard_index_env, -1); in ShouldShard() local
4280 if (total_shards == -1 && shard_index == -1) { in ShouldShard()
4282 } else if (total_shards == -1 && shard_index != -1) { in ShouldShard()
4285 << kTestShardIndex << " = " << shard_index in ShouldShard()
4290 } else if (total_shards != -1 && shard_index == -1) { in ShouldShard()
4298 } else if (shard_index < 0 || shard_index >= total_shards) { in ShouldShard()
4302 << ", but you have " << kTestShardIndex << "=" << shard_index in ShouldShard()
4333 bool ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) { in ShouldRunTestOnShard() argument
[all …]
|
| /external/llvm/utils/unittest/googletest/src/ |
| D | gtest.cc | 2702 const Int32 shard_index = Int32FromEnvOrDie(kTestShardIndex, -1); in OnTestIterationStart() local
2705 static_cast<int>(shard_index) + 1, in OnTestIterationStart()
4281 const Int32 shard_index = Int32FromEnvOrDie(shard_index_env, -1); in ShouldShard() local
4283 if (total_shards == -1 && shard_index == -1) { in ShouldShard()
4285 } else if (total_shards == -1 && shard_index != -1) { in ShouldShard()
4288 << kTestShardIndex << " = " << shard_index in ShouldShard()
4293 } else if (total_shards != -1 && shard_index == -1) { in ShouldShard()
4301 } else if (shard_index < 0 || shard_index >= total_shards) { in ShouldShard()
4305 << ", but you have " << kTestShardIndex << "=" << shard_index in ShouldShard()
4336 bool ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) { in ShouldRunTestOnShard() argument
[all …]
|
| D | gtest-internal-inl.h | 269 int total_shards, int shard_index, int test_id);
|
| /external/google-breakpad/src/testing/gtest/src/ |
| D | gtest.cc | 2729 const Int32 shard_index = Int32FromEnvOrDie(kTestShardIndex, -1); in OnTestIterationStart() local
2732 static_cast<int>(shard_index) + 1, in OnTestIterationStart()
4343 const Int32 shard_index = Int32FromEnvOrDie(shard_index_env, -1); in ShouldShard() local
4345 if (total_shards == -1 && shard_index == -1) { in ShouldShard()
4347 } else if (total_shards == -1 && shard_index != -1) { in ShouldShard()
4350 << kTestShardIndex << " = " << shard_index in ShouldShard()
4355 } else if (total_shards != -1 && shard_index == -1) { in ShouldShard()
4363 } else if (shard_index < 0 || shard_index >= total_shards) { in ShouldShard()
4367 << ", but you have " << kTestShardIndex << "=" << shard_index in ShouldShard()
4398 bool ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) { in ShouldRunTestOnShard() argument
[all …]
|
| D | gtest-internal-inl.h | 273 int total_shards, int shard_index, int test_id);
|
| /external/google-breakpad/src/testing/gtest/test/ |
| D | gtest_filter_unittest.py | 215 def RunWithSharding(total_shards, shard_index, command): argument
218 extra_env = {SHARD_INDEX_ENV_VAR: str(shard_index),
|
| D | gtest_unittest.cc | 2196 for (int shard_index = 0; shard_index < num_shards; shard_index++) { in TEST() local
2197 if (ShouldRunTestOnShard(num_shards, shard_index, test_id)) { in TEST()
2199 prev_selected_shard_index = shard_index; in TEST()
2202 << shard_index << " are both selected to run test " << test_id; in TEST()
2210 for (int shard_index = 0; shard_index < num_shards; shard_index++) { in TEST() local
2214 ShouldRunTestOnShard(num_shards, shard_index, test_id); in TEST()
|
| /external/googletest/googletest/test/ |
| D | googletest-filter-unittest.py | 220 def RunWithSharding(total_shards, shard_index, command): argument
223 extra_env = {SHARD_INDEX_ENV_VAR: str(shard_index),
|
| D | gtest_unittest.cc | 1965 for (int shard_index = 0; shard_index < num_shards; shard_index++) { in TEST() local
1966 if (ShouldRunTestOnShard(num_shards, shard_index, test_id)) { in TEST()
1968 prev_selected_shard_index = shard_index; in TEST()
1971 << shard_index << " are both selected to run test " << test_id; in TEST()
1979 for (int shard_index = 0; shard_index < num_shards; shard_index++) { in TEST() local
1983 ShouldRunTestOnShard(num_shards, shard_index, test_id); in TEST()
|
| /external/libaom/libaom/third_party/googletest/src/googletest/src/ |
| D | gtest.cc | 3072 const Int32 shard_index = Int32FromEnvOrDie(kTestShardIndex, -1); in OnTestIterationStart() local
3075 static_cast<int>(shard_index) + 1, in OnTestIterationStart()
4724 const Int32 shard_index = Int32FromEnvOrDie(shard_index_env, -1); in ShouldShard() local
4726 if (total_shards == -1 && shard_index == -1) { in ShouldShard()
4728 } else if (total_shards == -1 && shard_index != -1) { in ShouldShard()
4731 << kTestShardIndex << " = " << shard_index in ShouldShard()
4736 } else if (total_shards != -1 && shard_index == -1) { in ShouldShard()
4744 } else if (shard_index < 0 || shard_index >= total_shards) { in ShouldShard()
4748 << ", but you have " << kTestShardIndex << "=" << shard_index in ShouldShard()
4779 bool ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) { in ShouldRunTestOnShard() argument
[all …]
|
| D | gtest-internal-inl.h | 276 int total_shards, int shard_index, int test_id);
|
| /external/mesa3d/src/gtest/src/ |
| D | gtest.cc | 3071 const Int32 shard_index = Int32FromEnvOrDie(kTestShardIndex, -1); in OnTestIterationStart() local
3074 static_cast<int>(shard_index) + 1, in OnTestIterationStart()
4723 const Int32 shard_index = Int32FromEnvOrDie(shard_index_env, -1); in ShouldShard() local
4725 if (total_shards == -1 && shard_index == -1) { in ShouldShard()
4727 } else if (total_shards == -1 && shard_index != -1) { in ShouldShard()
4730 << kTestShardIndex << " = " << shard_index in ShouldShard()
4735 } else if (total_shards != -1 && shard_index == -1) { in ShouldShard()
4743 } else if (shard_index < 0 || shard_index >= total_shards) { in ShouldShard()
4747 << ", but you have " << kTestShardIndex << "=" << shard_index in ShouldShard()
4778 bool ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) { in ShouldRunTestOnShard() argument
[all …]
|
| D | gtest-internal-inl.h | 276 int total_shards, int shard_index, int test_id);
|
| /external/swiftshader/third_party/llvm-7.0/llvm/utils/unittest/googletest/src/ |
| D | gtest.cc | 3071 const Int32 shard_index = Int32FromEnvOrDie(kTestShardIndex, -1); in OnTestIterationStart() local
3074 static_cast<int>(shard_index) + 1, in OnTestIterationStart()
4723 const Int32 shard_index = Int32FromEnvOrDie(shard_index_env, -1); in ShouldShard() local
4725 if (total_shards == -1 && shard_index == -1) { in ShouldShard()
4727 } else if (total_shards == -1 && shard_index != -1) { in ShouldShard()
4730 << kTestShardIndex << " = " << shard_index in ShouldShard()
4735 } else if (total_shards != -1 && shard_index == -1) { in ShouldShard()
4743 } else if (shard_index < 0 || shard_index >= total_shards) { in ShouldShard()
4747 << ", but you have " << kTestShardIndex << "=" << shard_index in ShouldShard()
4778 bool ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) { in ShouldRunTestOnShard() argument
[all …]
|
| D | gtest-internal-inl.h | 276 int total_shards, int shard_index, int test_id);
|
| /external/libvpx/libvpx/third_party/googletest/src/src/ |
| D | gtest.cc | 3140 const Int32 shard_index = Int32FromEnvOrDie(kTestShardIndex, -1); in OnTestIterationStart() local
3143 static_cast<int>(shard_index) + 1, in OnTestIterationStart()
5265 const Int32 shard_index = Int32FromEnvOrDie(shard_index_env, -1); in ShouldShard() local
5267 if (total_shards == -1 && shard_index == -1) { in ShouldShard()
5269 } else if (total_shards == -1 && shard_index != -1) { in ShouldShard()
5272 << kTestShardIndex << " = " << shard_index in ShouldShard()
5277 } else if (total_shards != -1 && shard_index == -1) { in ShouldShard()
5285 } else if (shard_index < 0 || shard_index >= total_shards) { in ShouldShard()
5289 << ", but you have " << kTestShardIndex << "=" << shard_index in ShouldShard()
5320 bool ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) { in ShouldRunTestOnShard() argument
[all …]
|
| D | gtest-internal-inl.h | 272 int total_shards, int shard_index, int test_id);
|
| /external/googletest/googletest/src/ |
| D | gtest.cc | 3134 const Int32 shard_index = Int32FromEnvOrDie(kTestShardIndex, -1); in OnTestIterationStart() local
3137 static_cast<int>(shard_index) + 1, in OnTestIterationStart()
5331 const Int32 shard_index = Int32FromEnvOrDie(shard_index_env, -1); in ShouldShard() local
5333 if (total_shards == -1 && shard_index == -1) { in ShouldShard()
5335 } else if (total_shards == -1 && shard_index != -1) { in ShouldShard()
5338 << kTestShardIndex << " = " << shard_index in ShouldShard()
5343 } else if (total_shards != -1 && shard_index == -1) { in ShouldShard()
5351 } else if (shard_index < 0 || shard_index >= total_shards) { in ShouldShard()
5355 << ", but you have " << kTestShardIndex << "=" << shard_index in ShouldShard()
5386 bool ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) { in ShouldRunTestOnShard() argument
[all …]
|
| /external/libaom/libaom/test/ |
| D | test.cmake | 388 foreach(shard_index RANGE ${max_shard_index})
389 set(test_name "test_${shard_index}")
392 -DGTEST_SHARD_INDEX=${shard_index}
|
| /external/epid-sdk/ext/gtest/ |
| D | gtest-all.cc | 683 int total_shards, int shard_index, int test_id);
4523 const Int32 shard_index = Int32FromEnvOrDie(kTestShardIndex, -1); in OnTestIterationStart() local
4526 static_cast<int>(shard_index) + 1, in OnTestIterationStart()
6175 const Int32 shard_index = Int32FromEnvOrDie(shard_index_env, -1); in ShouldShard() local
6177 if (total_shards == -1 && shard_index == -1) { in ShouldShard()
6179 } else if (total_shards == -1 && shard_index != -1) { in ShouldShard()
6182 << kTestShardIndex << " = " << shard_index in ShouldShard()
6187 } else if (total_shards != -1 && shard_index == -1) { in ShouldShard()
6195 } else if (shard_index < 0 || shard_index >= total_shards) { in ShouldShard()
6199 << ", but you have " << kTestShardIndex << "=" << shard_index in ShouldShard()
[all …]
|
| /external/swiftshader/third_party/LLVM/utils/unittest/googletest/include/gtest/internal/ |
| D | gtest-internal-inl.h | 266 int total_shards, int shard_index, int test_id);
|