Home
last modified time | relevance | path

Searched defs:num_subchunks (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/v1/
Dall_reduce.py193 def _ring_permutations(num_workers, num_subchunks, gpu_perm): argument
254 def build_ring_all_reduce(input_tensors, num_workers, num_subchunks, argument
297 def _build_ring_gather(input_tensors, devices, num_subchunks, argument