Switch to new tf.distribute.Strategy reduce API. No longer allow a list
as a reduce or broadcast destination, since it isn't used outside of tests and will make future support for model parallelism more tricky. PiperOrigin-RevId: 223246154
Loading
Please sign in to comment