jainris commented on a change in pull request #6303:
URL: https://github.com/apache/incubator-tvm/pull/6303#discussion_r477160005
##########
File path: include/tvm/topi/transform.h
##########
@@ -1511,6 +1511,35 @@ inline Tensor sparse_to_dense(const Tensor&
sparse_indices, const Array<Integer>
name, tag);
}
+/*!
+ * \brief Returns a tensor with the diagonal of input tensor replaced with the
provided diagonal.
+ * \param input input tensor.
+ * \param diagonal values to be filled in the diagonal.
+ * \param name output tensor name.
+ * \param tag output tensor tag.
+ * \return new tensor with given diagonal values.
+ */
+inline Tensor matrix_set_diag(const Tensor& input, const Tensor& diagonal,
Review comment:
That might take some time.
Would it be fine to have that in a follow-up PR?
----------------------------------------------------------------
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:
us...@infra.apache.org
