Revert "refine ZeroGradFunctor in activation_op.h"

This reverts commit 1eeb11ef61.
port
sneaxiy 7 years ago
parent 64292f07d2
commit 748e204eff

@ -353,7 +353,7 @@ struct ZeroGradFunctor : 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) = out.constant(static_cast<T>(0));
dx.device(d) = static_cast<T>(0) / out;
}
};

Loading…
Cancel
Save