From c8d0b125f62a3b8785494f53d013809f8e7c8c29 Mon Sep 17 00:00:00 2001 From: Martin Wicke Date: Mon, 19 Mar 2018 17:50:05 -0700 Subject: [PATCH] Register gradient for argmin (cf. #15278). PiperOrigin-RevId: 189671974 --- tensorflow/python/ops/math_grad.py | 6 ++++++ 1 file changed, 6 insertions(+) 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.""" -- 2.7.4