Commit e513d1f5 authored by Benoit Steiner's avatar Benoit Steiner Committed by TensorFlower Gardener
Browse files

Materialize BroadcastGradientArgs by default instead of just doing so in... 

Materialize BroadcastGradientArgs by default instead of just doing so in aggressive mode. This ensures that we optimize gradient computations in the presence of variable batch sizes.

PiperOrigin-RevId: 181242749
parent e523d937
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