lhutton1 opened a new pull request #9576:
URL: https://github.com/apache/tvm/pull/9576
Supports legalizing a Relay mean operation to an equivalent series of NPU
operations. Mean can be legalized given one of three cases:
- Case 1 (axis == [1, 2] and keepsdims == True):
depthwise_conv2d + binary_elementwise
- Case 2 (ifm qparams == ofm qparams):
pooling
- Case 3 (else):
depthwise_conv2d
cc @ekalda @NicolaLancellotti @dchauhan-arm @manupa-arm @mbaret @jainris
Co-authored-by: Rishabh Jain <[email protected]>
--
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.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
