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

Replace call to gen_math_ops.add() with _add_dispatch() in reduce_logsumexp().

Browse Source 
PiperOrigin-RevId: 331568939
Change-Id: I046ae58c329b1b990d5e81282eb048203336439f
This commit is contained in:
A. Unique TensorFlower 2020-09-14 09:59:41 -07:00 committed by TensorFlower Gardener
parent 50e402dc00
commit 835f274efb
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/ops/math_ops.py
View File

@ -3046,7 +3046,7 @@ def reduce_logsumexp(input_tensor, axis=None, keepdims=False, name=None):
dims=reduce_dim))
if not keepdims:
my_max = array_ops.reshape(my_max, gen_array_ops.shape(result))
result = gen_math_ops.add(result, my_max)
result = _add_dispatch(result, my_max, name=name)
return _may_reduce_to_scalar(keepdims, axis, result)