Searched refs:per_replica_features (Results 1 – 1 of 1) sorted by relevance
1092 def sum_batch(per_replica_features): argument1103 return nest.map_structure(map_fn, per_replica_features)1235 def sum_batch(per_replica_features): argument1246 return nest.map_structure(map_fn, per_replica_features)