aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/cc
diff options
context:
space:
mode:
authorGravatar Yan Facai (颜发才) <facai.yan@gmail.com>2018-05-05 19:37:32 +0800
committerGravatar Yan Facai (颜发才) <facai.yan@gmail.com>2018-05-06 15:13:17 +0800
commitc8ebbd090d92ef6a49fa9f0f5a06952333beb2c8 (patch)
tree8e8443d9a070e6e44a85e1503e11fe521261fb5b /tensorflow/cc
parentd7b9e6c5a953151e4b83205d5741ab08723734a1 (diff)
CLN: fix minor error
Diffstat (limited to 'tensorflow/cc')
-rw-r--r--tensorflow/cc/gradients/math_grad.cc2
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/cc/gradients/math_grad.cc b/tensorflow/cc/gradients/math_grad.cc
index ea86fc0a7c..a8909846c9 100644
--- a/tensorflow/cc/gradients/math_grad.cc
+++ b/tensorflow/cc/gradients/math_grad.cc
@@ -453,7 +453,7 @@ Status UnsafeDivGrad(const Scope& scope, const Operation& op,
UnsafeDiv(scope, UnsafeDiv(scope, Neg(scope, x_1), x_2), x_2));
return BinaryGradCommon(scope, op, grad_outputs, gx_1, gx_2);
}
-REGISTER_GRADIENT_OP("UnsafeDiv", DivGrad);
+REGISTER_GRADIENT_OP("UnsafeDiv", UnsafeDivGrad);
Status SquaredDifferenceGrad(const Scope& scope, const Operation& op,
const std::vector<Output>& grad_inputs,