Searched refs:_per_device_aggregate_batch (Results 1 – 2 of 2) sorted by relevance
301 batch_outs = distributed_training_utils._per_device_aggregate_batch(
867 def _per_device_aggregate_batch(batch_outs, model, mode): function