Searched refs:value_flat (Results 1 – 1 of 1) sorted by relevance
3121 def batch_all_reduce(strategy, *value_flat): argument3123 reduce_op, [(v, _batch_reduce_destination(v)) for v in value_flat],3288 def batch_all_gather(strategy, *value_flat): argument3290 [(v, _batch_reduce_destination(v)) for v in value_flat], axis,