experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #40101 from gfkeith:norm_overflow

Browse Source 
PiperOrigin-RevId: 315371134
Change-Id: I58d1d40c83d25d23d2e8cfeacb120973fb25b928
This commit is contained in:
TensorFlower Gardener 2020-06-08 15:59:51 -07:00
commit 031d85fc24
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/python/keras/layers/preprocessing/normalization.py
View File

@ -177,7 +177,7 @@ class _NormalizingCombiner(Combiner):
reduction_counts = np.delete(values.shape, self.axis)
# We get the number of elements that will be reduced by multiplying all
# values of 'shape' corresponding to the reduced axes.
count = np.prod(reduction_counts, dtype=np.int32)
count = np.prod(reduction_counts, dtype=np.int64)
# We want to reduce across dimensions except those specified in 'axis'
# when using np.mean or np.variance; create the tuple of axes to reduce