fix type conversion problem in rsqrt functor

6 years ago · 91ba75000c
parent 082822d417
commit 91ba75000c
1 changed files with 1 additions and 1 deletions
--- a/paddle/fluid/operators/activation_op.h
+++ b/paddle/fluid/operators/activation_op.h
@ -477,7 +477,7 @@ struct RsqrtGradFunctor : public BaseActivationFunctor<T> {
  template <typename Device, typename X, typename Out, typename dOut,
            typename dX>
  void operator()(Device d, X x, Out out, dOut dout, dX dx) const {
-    dx.device(d) = static_cast<T>(-0.5) * dout * out.pow(3);
+    dx.device(d) = static_cast<T>(-0.5) * dout * out * out * out;
  }
 };