Searched refs:reduced_shards (Results 1 – 1 of 1) sorted by relevance
465 reduced_shards = _build_recursive_hd_gather(input_tensors, devices, red_op)467 reduced_shards = [un_op(t) for t in reduced_shards]468 output_tensors = _build_recursive_hd_scatter(reduced_shards, devices)578 reduced_shards = _build_shuffle_gather(input_tensors, gather_devices,580 output_tensors = _build_shuffle_scatter(reduced_shards, dst_devices)612 reduced_shards = []619 reduced_shards.append(red_shard)620 return reduced_shards623 def _build_shuffle_scatter(reduced_shards, dst_devices): argument638 out_tensors.append(array_ops.concat(reduced_shards, 0))[all …]