Home
last modified time | relevance | path

Searched refs:chunks_by_dev (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dall_reduce.py283 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_len
359 def _apply_unary_to_chunks(f, chunks_by_dev): argument
[all …]