Searched refs:reduced_shards (Results 1 – 1 of 1) sorted by relevance
471 reduced_shards = _build_recursive_hd_gather(input_tensors, devices, red_op)473 reduced_shards = [un_op(t) for t in reduced_shards]474 output_tensors = _build_recursive_hd_scatter(reduced_shards, devices)584 reduced_shards = _build_shuffle_gather(input_tensors, gather_devices,586 output_tensors = _build_shuffle_scatter(reduced_shards, dst_devices)618 reduced_shards = []625 reduced_shards.append(red_shard)626 return reduced_shards629 def _build_shuffle_scatter(reduced_shards, dst_devices): argument644 out_tensors.append(array_ops.concat(reduced_shards, 0))[all …]