Searched refs:_per_replica_aggregate_batch (Results 1 – 2 of 2) sorted by relevance
330 distributed_training_utils_v1._per_replica_aggregate_batch(
1040 def _per_replica_aggregate_batch(strategy, batch_outs, model, mode): function