Searched defs:value_on_replica (Results 1 – 1 of 1) sorted by relevance
70 def _gather_same_shape_and_verify(self, value_on_replica, axis, pure_eager, argument270 def _all_gather_same_shape_and_verify(self, value_on_replica, axis, argument