ANSHUMAN87 commented on a change in pull request #6685:
URL: https://github.com/apache/incubator-tvm/pull/6685#discussion_r508589066
##########
File path: python/tvm/relay/frontend/tensorflow.py
##########
@@ -890,6 +890,44 @@ def _impl(inputs, attr, params, mod):
return _impl
+def _sparse_tensor_dense_matmul():
+ # Sparse utility from Numpy
+ from scipy import sparse
+
+ def _impl(inputs, attr, params, mod):
+ assert len(inputs) == 4, "There should be 4 input tensors"
+
+ indices_tensor = _infer_value(inputs[0], params, mod).asnumpy()
+ values_tensor = _infer_value(inputs[1], params, mod).asnumpy()
+ dense_shape_tensor = _infer_value(inputs[2], params, mod).asnumpy()
+
+ data = inputs[3]
+
+ rows = [x[0] for x in indices_tensor]
+ cols = [x[1] for x in indices_tensor]
+
+ # Create Numpy sparse Tensor(CSR)
+ weight_sp = sparse.csr_matrix(
+ (values_tensor, (rows, cols)),
shape=tuple(dense_shape_tensor.tolist())
Review comment:
I totally agree with your comment, but as this PR is just an initial PR,
i feel lets keep the code as it is, it provides better readability in terms of
steps involved, later on, once all features are merged, we can work together to
optimize it :slightly_smiling_face:
Please let me know, in case you think otherwise!
----------------------------------------------------------------
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]
