From ce51d597963bd3e83cbae0e0afbce03b55b4118a Mon Sep 17 00:00:00 2001
From: Brian Patton <bjp@google.com>
Date: Thu, 10 Dec 2020 18:38:10 -0800
Subject: [PATCH] Explicitly specify no gradient for XlaVariadicReduce.

PiperOrigin-RevId: 346909834
Change-Id: I9a7b9c81a3cb5245d668e15778384ac6136bf3b2
---
 tensorflow/compiler/tf2xla/python/xla.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tensorflow/compiler/tf2xla/python/xla.py b/tensorflow/compiler/tf2xla/python/xla.py
index f9d7181ffac..22d4d07088e 100644
--- a/tensorflow/compiler/tf2xla/python/xla.py
+++ b/tensorflow/compiler/tf2xla/python/xla.py
@@ -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,