Change comment

eric-haibin-lin · Aug 8, 2017 · e48d7e1 · e48d7e1
1 parent e112b29
commit e48d7e1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/operator/tensor/square_sum-inl.h b/src/operator/tensor/square_sum-inl.h
@@ -185,7 +185,7 @@ struct SquareSumRspGradKernel<req, 1> {
 template<int req>
 struct SquareSumRspGradKernel<req, 1, kRowSparseStorage> {
   /*!
-   * \param i index of out_grad_row_idx
+   * \param i index of igrad.data()
    * \param in_grad_row_idx row_idx of the gradient of the op's input
    * \param in_grad gradient of the op's input
    * \param out_grad_row_idx row_idx of the gradient of the op's output