Searched defs:per_replica_values (Results 1 – 3 of 3) sorted by relevance
646 def _group_value_by_device(per_replica_values): argument879 def _batch_all_reduce(self, reduce_op, per_replica_values): argument1187 def _all_reduce_per_replica_values(self, reduce_op, per_replica_values, argument1303 def _batch_all_gather(self, per_replica_values, axis, options): argument
1159 def map_fn(per_replica_values): argument1296 def map_fn(per_replica_values): argument
212 def flatten_per_replica_values(distribution_strategy, per_replica_values): argument