Searched defs:per_replica_value (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | cross_device_ops.py | 216 def _simple_reduce(per_replica_value, reduce_to_device, accumulation_fn, argument
236 def _simple_gather(per_replica_value, reduce_to_device, axis): argument
272 def reduce(self, reduce_op, per_replica_value, destinations, options=None): argument
320 def _gather(self, per_replica_value, destinations, axis, options=None): argument
366 def _gather_implementation(self, per_replica_value, destinations, axis, argument
466 def reduce_implementation(self, reduce_op, per_replica_value, destinations, argument
608 def reduce_implementation(self, reduce_op, per_replica_value, destinations, argument
623 def _gather_implementation(self, per_replica_value, destinations, axis, argument
856 def reduce_implementation(self, reduce_op, per_replica_value, destinations, argument
937 def _gather_implementation(self, per_replica_value, destinations, axis, argument
[all …]
|
| H A D | strategy_gather_test.py | 275 def replica_fn(per_replica_value): argument
347 def replica_fn(per_replica_value): argument
|