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

Explicitly specify no gradient for XlaVariadicReduce.

Browse Source 
PiperOrigin-RevId: 346909834
Change-Id: I9a7b9c81a3cb5245d668e15778384ac6136bf3b2
This commit is contained in:
Brian Patton 2020-12-10 18:38:10 -08:00 committed by TensorFlower Gardener
parent a2d2b5c3d5
commit ce51d59796
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/compiler/tf2xla/python/xla.py
View File

@ -342,6 +342,8 @@ recv = gen_xla_ops.xla_recv
reduce = gen_xla_ops.xla_reduce
variadic_reduce = gen_xla_ops.xla_variadic_reduce
ops.no_gradient("XlaVariadicReduce")
def reduce_window(operand,
init,