Searched defs:value_flat (Results 1 – 1 of 1) sorted by relevance
3121 def batch_all_reduce(strategy, *value_flat):3288 def batch_all_gather(strategy, *value_flat):