quic-sanirudh opened a new pull request #6138:
URL: https://github.com/apache/incubator-tvm/pull/6138
- This patch adds a new member to ReduceNode called init which allows
initialization with a custom ProducerLoad or a Float/Int immediate.
- This allows initialization of the output Tensor of a reduction with
another Tensor instead of the `identity_element` defined in the
CommReducer
- One example use case for this node is to initialize the Output of a
convolution reduction with the Bias values thereby saving the
Bias-add computation.
Thanks for contributing to TVM! Please refer to guideline
https://tvm.apache.org/docs/contribute/ for useful information and tips. After
the pull request is submitted, please request code reviews from
[Reviewers](https://github.com/apache/incubator-tvm/blob/master/CONTRIBUTORS.md#reviewers)
by @ them in the pull request thread.
----------------------------------------------------------------
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]
