Searched refs:group_sizes (Results 1 – 3 of 3) sorted by relevance
| /external/ComputeLibrary/src/runtime/ |
| D | BlobLifetimeManager.cpp | 71 std::vector<BlobInfo> group_sizes; in update_blobs_and_mappings() local
72 …std::transform(std::begin(_free_blobs), std::end(_free_blobs), std::back_inserter(group_sizes), []… in update_blobs_and_mappings()
78 size_t max_size = std::max(_blobs.size(), group_sizes.size()); in update_blobs_and_mappings()
80 group_sizes.resize(max_size); in update_blobs_and_mappings()
81 …std::transform(std::begin(_blobs), std::end(_blobs), std::begin(group_sizes), std::begin(_blobs), … in update_blobs_and_mappings()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | reduce_scatter_utils.cc | 450 std::vector<int64_t> group_sizes; in MatchReduceScatter() local
451 group_sizes.reserve(split_dims.size()); in MatchReduceScatter()
453 group_sizes.push_back(user->operand(0)->shape().dimensions(dim) / in MatchReduceScatter()
456 if (Product(group_sizes) != group_size) { in MatchReduceScatter()
471 group_sizes, ar)) { in MatchReduceScatter()
|
| /external/python/cpython3/Lib/test/ |
| D | test_statistics.py | 2385 group_sizes = {total // n, total // n + 1}
2388 self.assertTrue(sizes <= group_sizes)
|