aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/cc/gradients
diff options
context:
space:
mode:
authorGravatar Suharsh Sivakumar <suharshs@google.com>2017-02-15 15:34:07 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-02-15 15:48:19 -0800
commitd3a41971ccc7c03e8b476a48dc8aa7fbb983431c (patch)
tree07879daa201aeb9c58485042a6f30d846c75bd54 /tensorflow/cc/gradients
parent9f12dd9bf44797abe1a22e4520d405704d133d9c (diff)
Make input_min and input_max Inputs of QuantizeAndDequantize rather than attributes.
Change: 147653779
Diffstat (limited to 'tensorflow/cc/gradients')
-rw-r--r--tensorflow/cc/gradients/array_grad.cc1
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/cc/gradients/array_grad.cc b/tensorflow/cc/gradients/array_grad.cc
index dd53236976..cafa15347e 100644
--- a/tensorflow/cc/gradients/array_grad.cc
+++ b/tensorflow/cc/gradients/array_grad.cc
@@ -88,6 +88,7 @@ Status QuantizeAndDequantizeGrad(const Scope& scope, const Operation& op,
return scope.status();
}
REGISTER_GRADIENT_OP("QuantizeAndDequantize", QuantizeAndDequantizeGrad);
+REGISTER_GRADIENT_OP("QuantizeAndDequantizeV2", QuantizeAndDequantizeGrad);
Status SplitGrad(const Scope& scope, const Operation& op,
const std::vector<Output>& grad_inputs,