Searched refs:batch_reduce_to (Results 1 – 7 of 7) sorted by relevance
38 name: "batch_reduce_to"
1164 def batch_reduce_to(self, reduce_op, value_destination_pairs): member in DistributionStrategyExtended1502 return strategy.extended.batch_reduce_to(
415 reduced_grads = distribution.extended.batch_reduce_to(
563 return test_object.extended.batch_reduce_to(reduce_op,
914 reduced_grads = distribution.extended.batch_reduce_to(
644 reduced_grads = distribution.extended.batch_reduce_to(