From: Martin Wicke <wicke@google.com>
Date: Tue, 20 Mar 2018 00:50:05 +0000 (-0700)
Subject: Register gradient for argmin (cf. #15278).
X-Git-Tag: tflite-v0.1.7~145^2^2~39
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=c8d0b125f62a3b8785494f53d013809f8e7c8c29;p=platform%2Fupstream%2Ftensorflow.git

Register gradient for argmin (cf. #15278).

PiperOrigin-RevId: 189671974
---

diff --git a/tensorflow/python/ops/math_grad.py b/tensorflow/python/ops/math_grad.py
index eb33687..02e07dc 100644
--- a/tensorflow/python/ops/math_grad.py
+++ b/tensorflow/python/ops/math_grad.py
@@ -41,6 +41,12 @@ def _ArgMaxGrad(op, grad):
   return [None, None]
 
 
+@ops.RegisterGradient("ArgMin")
+def _ArgMinGrad(op, grad):
+  del op, grad
+  return [None, None]
+
+
 @ops.RegisterGradient("Sum")
 def _SumGrad(op, grad):
   """Gradient for Sum."""