diff options
| author |  KB Sriram <kbsriram@gmail.com> | 2018-07-21 07:57:44 -0700 |
|---|---|---|
| committer | KB Sriram <kbsriram@gmail.com> | 2018-07-21 08:15:35 -0700 |
| commit | a2c479b6fd1ec75f37b8104f53a9f4a286cb72ce (patch) | |
| tree | fd50d98e9272d7e19d59266b972e33fd6f531894 /tensorflow/cc/framework | |
| parent | dae7a75734f2137aae7130e064fab9dfcb799c45 (diff) | |
Add C++ gradients for some image operators.
Added gradients and tests for
- ResizeBilinear
- ResizeBicubic
- ResizeNearestNeighbor
Note: Some of the tests are for the operator itself rather than
the gradient, paralleling existing tests in image_grad.py
See https://github.com/tensorflow/tensorflow/issues/21019
Diffstat (limited to 'tensorflow/cc/framework')
| -rw-r--r-- | tensorflow/cc/framework/gradient_checker.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/cc/framework/gradient_checker.cc b/tensorflow/cc/framework/gradient_checker.cc index de2645cb44..695180c23b 100644 --- a/tensorflow/cc/framework/gradient_checker.cc +++ b/tensorflow/cc/framework/gradient_checker.cc @@ -409,6 +409,7 @@ Status ComputeGradientError(const Scope& scope, const Output& x, const Output& y, const TensorShape& y_shape, JAC_T* max_error); INSTANTIATE_GRAD_ERR_TYPE(float, float, float); +INSTANTIATE_GRAD_ERR_TYPE(double, float, double); INSTANTIATE_GRAD_ERR_TYPE(double, double, double); INSTANTIATE_GRAD_ERR_TYPE(complex64, float, float); INSTANTIATE_GRAD_ERR_TYPE(float, complex64, float); |