Searched defs:per_replica_value (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_ops.py | 218 def _simple_reduce(per_replica_value, reduce_to_device, accumulation_fn, argument
238 def _simple_gather(per_replica_value, reduce_to_device, axis): argument
274 def reduce(self, reduce_op, per_replica_value, destinations, options=None): argument
322 def _gather(self, per_replica_value, destinations, axis, options=None): argument
369 def _gather_implementation(self, per_replica_value, destinations, axis, argument
469 def reduce_implementation(self, reduce_op, per_replica_value, destinations, argument
611 def reduce_implementation(self, reduce_op, per_replica_value, destinations, argument
626 def _gather_implementation(self, per_replica_value, destinations, axis, argument
859 def reduce_implementation(self, reduce_op, per_replica_value, destinations, argument
940 def _gather_implementation(self, per_replica_value, destinations, axis, argument
[all …]
|
| D | strategy_gather_test.py | 282 def replica_fn(per_replica_value): argument
354 def replica_fn(per_replica_value): argument
|