Searched refs:new_chunks (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/v1/ |
| D | all_reduce.py | 504 new_chunks = [[] for _ in devices]
514 new_chunks[d] = red_op(left_split[0], right_split[0])
516 new_chunks[d + span] = red_op(left_split[1], right_split[1])
517 chunks = new_chunks
539 new_chunks = [[] for _ in devices]
549 new_chunks[left_idx] = array_ops.concat([chunks[left_idx],
552 new_chunks[right_idx] = array_ops.concat([chunks[left_idx],
554 chunks = new_chunks
|
| /external/llvm-project/openmp/runtime/src/ |
| D | kmp_dispatch_hier.h | 634 const ST *new_chunks) const { in need_to_reallocate()
643 if (info[i].chunk != new_chunks[i]) in need_to_reallocate()
654 const enum sched_type *new_scheds, const ST *new_chunks) { in allocate_hier()
656 if (!need_to_reallocate(n, new_layers, new_scheds, new_chunks)) { in allocate_hier()
681 info[i].chunk = new_chunks[i]; in allocate_hier()
917 typename traits_t<T>::signed_t *new_chunks, in __kmp_dispatch_init_hierarchy() argument
935 gtid, i, layer, i, (int)new_scheds[i], i, new_chunks[i])); in __kmp_dispatch_init_hierarchy()
941 KMP_DEBUG_ASSERT(new_chunks); in __kmp_dispatch_init_hierarchy()
980 sh->hier->allocate_hier(n, new_layers, new_scheds, new_chunks); in __kmp_dispatch_init_hierarchy()
|
| /external/mesa3d/src/gallium/winsys/amdgpu/drm/ |
| D | amdgpu_bo.c | 988 struct amdgpu_sparse_backing_chunk *new_chunks = in sparse_backing_free() local
992 if (!new_chunks) in sparse_backing_free()
996 backing->chunks = new_chunks; in sparse_backing_free()
|
| /external/rust/crates/itertools/src/ |
| D | groupbylazy.rs | 430 pub fn new_chunks<J>(iter: J, size: usize) -> IntoChunks<J::IntoIter> in new_chunks() function
|
| D | lib.rs | 593 groupbylazy::new_chunks(self, size) in chunks()
|