Searched refs:value_flat (Results 1 – 1 of 1) sorted by relevance
1501 def batch_all_reduce(strategy, *value_flat): argument1503 reduce_op, [(v, _batch_reduce_destination(v)) for v in value_flat])