ANSHUMAN87 commented on a change in pull request #6889:
URL: https://github.com/apache/tvm/pull/6889#discussion_r538482623
##########
File path: python/tvm/relay/op/nn/nn.py
##########
@@ -1993,17 +1993,27 @@ def batch_matmul(x, y):
return _make.batch_matmul(x, y)
-def sparse_dense(data, weight):
+# pylint: disable=no-else-return,inconsistent-return-statements
+def sparse_dense(dense_mat, sparse_mat, sparse_lhs=False):
r"""
- Computes the matrix multiplication of `data` and `weight`, where `data` is
- a dense matrix and `weight` is a sparse (either BSR or CSR) namedtuple with
+ Computes the matrix multiplication of `dense_mat` and `sparse_mat`, where
`dense_mat` is
+ a dense matrix and `sparse_mat` is a sparse (either BSR or CSR) namedtuple
with
fields `data`, `indices`, and `indptr`.
.. math::
- \mbox{sparse_dense}(data, weight)[m, n] = \mbox{matmul}(x,
\mbox{as_dense}(weight)^T)[m, n]
+ if sparse_lhs=True
Review comment:
@tkonolige : You are right the rendering does not look good :)
With new change the new rendering looks as below:
I hope this is okay.
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
