diff options
| author |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-03 17:31:50 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-03 17:31:50 -0700 |
| commit | c4b12fbd0341676375f97f2be8a94d3bfadfb70e (patch) | |
| tree | 19b10c46047d1942ca48e6cf6da780edb3dcd7c9 /tensorflow/cc/framework | |
| parent | f10b2802611be27fcfcefbcfeb508402b44aad8d (diff) | |
| parent | a2c479b6fd1ec75f37b8104f53a9f4a286cb72ce (diff) | |
Merge pull request #21021 from kbsriram:tf-image-grad
PiperOrigin-RevId: 207350813
Diffstat (limited to 'tensorflow/cc/framework')
| -rw-r--r-- | tensorflow/cc/framework/gradient_checker.cc | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/cc/framework/gradient_checker.cc b/tensorflow/cc/framework/gradient_checker.cc index de2645cb44..a1eb0d9d08 100644 --- a/tensorflow/cc/framework/gradient_checker.cc +++ b/tensorflow/cc/framework/gradient_checker.cc @@ -247,7 +247,7 @@ Status ComputeNumericJacobianTranspose(const Scope& scope, const OutputList& xs, auto y_pos_flat = y_pos[y_idx].flat<Y_T>(); auto y_neg_flat = y_neg[y_idx].flat<Y_T>(); const int64 y_size = y_shapes[y_idx].num_elements(); - const Y_T scale = Y_T{2 * delta}; + const Y_T scale = 2 * delta; auto jacobian = (*jacobian_ts)[x_idx * y_num + y_idx].matrix<JAC_T>(); for (int c = 0; c < y_size; ++c) { SetJacobian<Y_T, JAC_T>(&jacobian, r * x_stride + unit_dimension, @@ -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); |