locnd182644 opened a new pull request, #18658:
URL: https://github.com/apache/tvm/pull/18658
This PR support handle slope and axis argument of PReLu op with different
slope shapes: (1xCx1x1) or (S,) or (1,1) etc.
### Description
- Handle slope and axis argument of PReLu op (to pass into relax.op.nn.prelu
function)
- If slope shape = (1xCx1x1), get axis = 1 and reshape slope to (C,)
- else if slope shape = (S,) or (1, 1), get axis = len(x_shape) - 1 (take
the last axis of the input x) (https://onnx.ai/onnx/repo-docs/Broadcasting.html)
- else raise error
### Resolved
- Fixed 1: #18596
- Fixed 2: #18598
- Fixed 3: #18606
- Fixed 4: #18607
--
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]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
