Searched defs:reduce_to_device (Results 1 – 1 of 1) sorted by relevance
204 def _simple_reduce(per_replica_value, reduce_to_device, accumulation_fn, argument369 def __init__(self, reduce_to_device=None, accumulation_fn=None): argument