Searched defs:per_replica_values (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_ops.py | 626 def _group_value_by_device(per_replica_values): argument
859 def _batch_all_reduce(self, reduce_op, per_replica_values): argument
1181 def _batch_all_reduce(self, reduce_op, per_replica_values, options): argument
1198 def _do_batch_all_reduce_dense(self, reduce_op, per_replica_values, options): argument
1266 def _do_batch_all_reduce_sparse(self, reduce_op, per_replica_values, options): argument
1332 def _batch_all_gather(self, per_replica_values, axis, options): argument
|
| D | input_lib_test.py | 1095 def map_fn(per_replica_values): argument
1238 def map_fn(per_replica_values): argument
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | distributed_training_utils_v1.py | 216 def flatten_per_replica_values(distribution_strategy, per_replica_values): argument
|