Commit 67593ee4 authored by Priya Gupta's avatar Priya Gupta Committed by TensorFlower Gardener
Browse files

Add a new enum for reduction type in distribution strategy. Currently supports... 

Add a new enum for reduction type in distribution strategy. Currently supports SUM, MEAN and ONLY_FIRST_REPLICA
Allow accepting the new enum in `reduce` and `batch_reduce` APIs.
Change some callers to use the new enum.

PiperOrigin-RevId: 221475510
parent 00774b2d
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment