diff --git a/src/operator/numpy/linalg/np_gesvd-inl.h b/src/operator/numpy/linalg/np_gesvd-inl.h index cbe65e2297ce..efcc65ae3f47 100644 --- a/src/operator/numpy/linalg/np_gesvd-inl.h +++ b/src/operator/numpy/linalg/np_gesvd-inl.h @@ -111,7 +111,9 @@ void NumpyLaGesvdForward(const nnvm::NodeAttrs& attrs, }); } -// Helper for gesvd_backward. See technical report for details +// Helper for gesvd_backward. See technical report +// `Auto-Differentiating Linear Algebra` for details +// on https://arxiv.org/pdf/1710.08717.pdf template DType gesvd_back_helper_eps(DType* X);