Commit 5973383f authored by Yunxing Dai's avatar Yunxing Dai Committed by TensorFlower Gardener
Browse files

Change batchnorm expander to use GetDimensionSize. 

This is necessary if the input of batchnorm has dynamic
dimension. Otherwise the mean calculation would be incorrect.

PiperOrigin-RevId: 223840830
parent f042706a
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