Searched defs:per_replica_values (Results 1 – 1 of 1) sorted by relevance
402 def _group_value_by_device(per_replica_values): argument694 def _batch_all_reduce(self, reduce_op, per_replica_values): argument878 def _batch_all_reduce(self, reduce_op, per_replica_values): argument1017 def _batch_all_reduce(self, reduce_op, per_replica_values): argument