Skip to content

Commit

Permalink
fix lint
Browse files Browse the repository at this point in the history
  • Loading branch information
yajiedesign committed Nov 30, 2015
1 parent 0c4da94 commit 2e1fd25
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion src/ndarray/unary_function-inl.h
Original file line number Diff line number Diff line change
Expand Up @@ -119,7 +119,8 @@ MXNET_REGISTER_TBLOB_FUN(sqrt, XPU)
// rsqrt
MXNET_REGISTER_TBLOB_FUN(rsqrt, XPU)
.set_function(XPU::kDevMask, UnaryForward_<XPU, op::mshadow_op::reciprocal_square_root>, true)
.set_gradient(XPU::kDevMask, UnaryBackwardUseIn_<XPU, op::mshadow_op::reciprocal_square_root_grad>, true)
.set_gradient(XPU::kDevMask,
UnaryBackwardUseIn_<XPU, op::mshadow_op::reciprocal_square_root_grad>, true)
.describe("Take rsqrt of the src");
// exp
MXNET_REGISTER_TBLOB_FUN(exp, XPU)
Expand Down

0 comments on commit 2e1fd25

Please sign in to comment.