diff options
Diffstat (limited to 'tensorflow/python/ops/math_grad.py')
| -rw-r--r-- | tensorflow/python/ops/math_grad.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/tensorflow/python/ops/math_grad.py b/tensorflow/python/ops/math_grad.py index 8d999f0074..2e8328a98b 100644 --- a/tensorflow/python/ops/math_grad.py +++ b/tensorflow/python/ops/math_grad.py @@ -796,6 +796,12 @@ def _FloorGrad(_, unused_grad): return [None] +@ops.RegisterGradient("Rint") +def _RintGrad(_, unused_grad): + # the gradient of Rint is zero + return [None] + + @ops.RegisterGradient("BatchMatMul") def _BatchMatMul(op, grad): """Returns the gradient of x and y given the gradient of x * y.""" |