eric-haibin-lin commented on a change in pull request #8938: Add operator for dot(dns, csr) = csr URL: https://github.com/apache/incubator-mxnet/pull/8938#discussion_r159513184
########## File path: src/operator/tensor/dot-inl.h ########## @@ -811,6 +897,96 @@ inline void DotCsrRspRspImpl(const OpContext& ctx, }); } +/* + * \brief CPU Impl of dot(dns, csr) = csr + */ +template<typename xpu> +inline void DotDnsCsrCsrImpl(const OpContext& ctx, + const TBlob& lhs, const NDArray& rhs, + const OpReqType req, NDArray* ret) { + if (kNullOp == req) return; + + CHECK_EQ(req, kWriteTo); + CHECK_EQ(rhs.storage_type(), kCSRStorage); + + using namespace mshadow; + using namespace mshadow::expr; + using nnvm::dim_t; + + /*Initialize data structures*/ Review comment: nit: space after /* ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org With regards, Apache Git Services