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

Fix reduction with a specified axis.

Browse Source 
PiperOrigin-RevId: 286185937
Change-Id: I56bed77a84b758f204888102e8a7d7175b6e3d9e
This commit is contained in:
Chris Jones 2019-12-18 07:16:34 -08:00 committed by TensorFlower Gardener
parent 6da1b29ae3
commit 949b18f0fe
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/distribute/distribute_lib.py
View File

@ -840,7 +840,7 @@ class Strategy(object):
if dim is not None: if dim is not None:
# By returning a python value in the static shape case, we can # By returning a python value in the static shape case, we can
# maybe get a fast path for reducing the denominator. # maybe get a fast path for reducing the denominator.
return numer, dim return numer, array_ops.constant(dim, dtype=dtypes.int64)
elif axis < 0: elif axis < 0:
axis = axis + array_ops.rank(v) axis = axis + array_ops.rank(v)
if v.shape.rank == 1: if v.shape.rank == 1: