From: Alexandre Passos <apassos@google.com>
Date: Sat, 24 Feb 2018 00:51:50 +0000 (-0800)
Subject: Registers None gradients for ArgMax
X-Git-Tag: upstream/v1.7.0~31^2~392
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=18bab99ac33f31192d400aebcfb7670a121655bd;p=platform%2Fupstream%2Ftensorflow.git

Registers None gradients for ArgMax

PiperOrigin-RevId: 186843686
---

diff --git a/tensorflow/python/ops/math_grad.py b/tensorflow/python/ops/math_grad.py
index 9d5289f..bf28f74 100644
--- a/tensorflow/python/ops/math_grad.py
+++ b/tensorflow/python/ops/math_grad.py
@@ -35,6 +35,12 @@ def _safe_shape_div(x, y):
   return x // math_ops.maximum(y, 1)
 
 
+@ops.RegisterGradient("ArgMax")
+def _ArgMaxGrad(op, grad):
+  del op, grad
+  return [None, None]
+
+
 @ops.RegisterGradient("Sum")
 def _SumGrad(op, grad):
   """Gradient for Sum."""