Searched refs:chunks_by_dev (Results 1 – 1 of 1) sorted by relevance
283 chunks_by_dev, pad_len = _build_ring_gather(287 chunks_by_dev = _apply_unary_to_chunks(un_op, chunks_by_dev)289 chunks_by_dev)328 chunks_by_dev = []333 chunks_by_dev.append(splits)347 chunks_by_dev[pred_dev][chunk_index],348 chunks_by_dev[d][chunk_index])355 chunks_by_dev[d][chunk_index] = new_partial_reductions[chunk_index]356 return chunks_by_dev, split_pad_len359 def _apply_unary_to_chunks(f, chunks_by_dev): argument[all …]