diff options
author | 2018-06-19 02:55:11 -0700 | |
---|---|---|
committer | 2018-06-19 02:57:55 -0700 | |
commit | a89726dea8d9005a5f9ca73ad14f28c32cd87e56 (patch) | |
tree | d653b7d65c2ec55b1632ddbb2612d1b0ac4ed3a2 /tensorflow/core/ops/math_ops.cc | |
parent | d091290a22aba19cf43a697c6194bb4da98ebae6 (diff) |
Derivative of tf.igamma(a, x) and tf.igammac(a, x) with respect to a.
Previously, both functions only supported the derivative with respect to x. We add the derivative with respect to the other argument. It is computed using the Eigen function igamma_der_a that performs forward-mode differentiation of the code for igamma. This function is not exposed in the public TensorFlow API.
PiperOrigin-RevId: 201145398
Diffstat (limited to 'tensorflow/core/ops/math_ops.cc')
-rw-r--r-- | tensorflow/core/ops/math_ops.cc | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/tensorflow/core/ops/math_ops.cc b/tensorflow/core/ops/math_ops.cc index b3487122e2..1681d63930 100644 --- a/tensorflow/core/ops/math_ops.cc +++ b/tensorflow/core/ops/math_ops.cc @@ -489,6 +489,13 @@ REGISTER_OP("Igamma") .Attr("T: {float, double}") .SetShapeFn(shape_inference::BroadcastBinaryOpShapeFn); +REGISTER_OP("IgammaGradA") + .Input("a: T") + .Input("x: T") + .Output("z: T") + .Attr("T: {float, double}") + .SetShapeFn(shape_inference::BroadcastBinaryOpShapeFn); + REGISTER_OP("Zeta") .Input("x: T") .Input("q: T") |