| /external/tensorflow/tensorflow/core/util/tensor_bundle/ |
| H A D | naming.cc | 28 string DataFilename(StringPiece prefix, int32 shard_id, int32 num_shards) { argument
29 DCHECK_GT(num_shards, 0);
30 DCHECK_LT(shard_id, num_shards);
33 shard_id, num_shards);
|
| H A D | tensor_bundle.cc | 531 int num_shards = 0; member in struct:tensorflow::MergeState
563 int num_shards; local
575 merge_state->num_shards += header.num_shards();
596 num_shards = header.num_shards();
639 {DataFilename(prefix, to_merge_entry.shard_id(), num_shards),
675 header.set_num_shards(merge.num_shards);
731 num_shards_ = header.num_shards();
|
| /external/tensorflow/tensorflow/core/lib/core/ |
| H A D | blocking_counter_test.cc | 57 const int num_shards = num_threads * shards_per_thread; local
60 BlockingCounter bc(num_shards);
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/ |
| H A D | parallel_for.cc | 30 const int num_shards = std::max<int>( local
32 const int64 block_size = (batch_size + num_shards - 1) / num_shards;
|
| /external/tensorflow/tensorflow/core/util/ |
| H A D | work_sharder.cc | 39 // We shard [0, total) into "num_shards" shards.
40 // 1 <= num_shards <= num worker threads
46 const int num_shards = local
56 const int64 block_size = (total + num_shards - 1) / num_shards;
|
| H A D | work_sharder_test.cc | 33 int64 num_shards = 0; local
37 [=, &mu, &num_shards, &num_done_work, &work](int64 start, int64 limit) {
42 ++num_shards;
51 << num_shards; local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| H A D | candidate_sampler_ops.cc | 171 int64 num_shards; local
172 OP_REQUIRES_OK(context, context->GetAttr("num_shards", &num_shards));
179 num_shards, shard));
182 num_reserved_ids, num_shards, shard));
|
| H A D | range_sampler.cc | 236 int32 num_shards, int32 shard)
239 num_shards_(num_shards),
252 int32 num_shards, int32 shard)
255 num_shards_(num_shards),
232 FixedUnigramSampler(Env* env, int64 range, const string& vocab_file, float distortion, int32 num_reserved_ids, int32 num_shards, int32 shard) argument
248 FixedUnigramSampler(int64 range, const std::vector<float>& unigrams, float distortion, int32 num_reserved_ids, int32 num_shards, int32 shard) argument
|
| H A D | topk_op_gpu.cu.cc | 262 // mergeShards performs a top-k merge on `num_shards` many sorted streams that
264 // |s_1 1st|s_2 1st|...s_{num_shards} 1st|s_1 2nd|s_2 2nd|...
269 __device__ void mergeShards(int num_shards, int k, argument
273 // If k < num_shards, we can use a min-heap with k elements to get the top k
275 // If k > num_shards, we can initialize a min-heap with the top element from
277 const int heap_size = k < num_shards ? k : num_shards;
290 // Now perform top k with the remaining shards (if num_shards > heap_size).
291 for (int shard = heap_size; shard < num_shards; shard++) {
323 int next_shard_index = shard_index + num_shards;
[all...] |
| /external/google-breakpad/src/testing/gtest/test/ |
| H A D | gtest_unittest.cc | 2191 const int num_shards = 5; local
2196 for (int shard_index = 0; shard_index < num_shards; shard_index++) {
2197 if (ShouldRunTestOnShard(num_shards, shard_index, test_id)) {
2210 for (int shard_index = 0; shard_index < num_shards; shard_index++) {
2214 ShouldRunTestOnShard(num_shards, shard_index, test_id);
2216 EXPECT_GE(num_tests_on_shard, num_tests / num_shards);
|
| /external/googletest/googletest/test/ |
| H A D | gtest_unittest.cc | 1891 const int num_shards = 5; local
1896 for (int shard_index = 0; shard_index < num_shards; shard_index++) {
1897 if (ShouldRunTestOnShard(num_shards, shard_index, test_id)) {
1910 for (int shard_index = 0; shard_index < num_shards; shard_index++) {
1914 ShouldRunTestOnShard(num_shards, shard_index, test_id);
1916 EXPECT_GE(num_tests_on_shard, num_tests / num_shards);
|
| /external/protobuf/gtest/test/ |
| H A D | gtest_unittest.cc | 2029 const int num_shards = 5; local
2034 for (int shard_index = 0; shard_index < num_shards; shard_index++) {
2035 if (ShouldRunTestOnShard(num_shards, shard_index, test_id)) {
2048 for (int shard_index = 0; shard_index < num_shards; shard_index++) {
2052 ShouldRunTestOnShard(num_shards, shard_index, test_id);
2054 EXPECT_GE(num_tests_on_shard, num_tests / num_shards);
|
| /external/v8/testing/gtest/test/ |
| H A D | gtest_unittest.cc | 1891 const int num_shards = 5; local
1896 for (int shard_index = 0; shard_index < num_shards; shard_index++) {
1897 if (ShouldRunTestOnShard(num_shards, shard_index, test_id)) {
1910 for (int shard_index = 0; shard_index < num_shards; shard_index++) {
1914 ShouldRunTestOnShard(num_shards, shard_index, test_id);
1916 EXPECT_GE(num_tests_on_shard, num_tests / num_shards);
|
| /external/vulkan-validation-layers/tests/gtest-1.7.0/test/ |
| H A D | gtest_unittest.cc | 1884 const int num_shards = 5; local
1889 for (int shard_index = 0; shard_index < num_shards; shard_index++) {
1890 if (ShouldRunTestOnShard(num_shards, shard_index, test_id)) {
1903 for (int shard_index = 0; shard_index < num_shards; shard_index++) {
1907 ShouldRunTestOnShard(num_shards, shard_index, test_id);
1909 EXPECT_GE(num_tests_on_shard, num_tests / num_shards);
|