Searched defs:per_replica_value (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_ops.py | 200 def _simple_reduce(per_replica_value, reduce_to_device, accumulation_fn, argument
220 def _simple_gather(per_replica_value, reduce_to_device, axis): argument
255 def reduce(self, reduce_op, per_replica_value, destinations, options=None): argument
303 def _gather(self, per_replica_value, destinations, axis, options=None): argument
350 def _gather_implementation(self, per_replica_value, destinations, axis, argument
450 def reduce_implementation(self, reduce_op, per_replica_value, destinations, argument
588 def reduce_implementation(self, reduce_op, per_replica_value, destinations, argument
603 def _gather_implementation(self, per_replica_value, destinations, axis, argument
836 def reduce_implementation(self, reduce_op, per_replica_value, destinations, argument
917 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
|