[GitHub] [tvm] 0x00-pl commented on pull request #11606: [ONNX] Add ReduceSum opset13 support (non-dynamic)

Wed, 29 Mar 2023 00:24:36 -0700 


0x00-pl commented on PR #11606:
URL: https://github.com/apache/tvm/pull/11606#issuecomment-1488075553

   ```
                   # Get axis and unpack scalar
                   constant_axis = int(inputs[1].data.numpy()[0])
   ```
   
https://github.com/apache/tvm/blame/b3a5e18f6f096c2828512ae56f2e40261350c8b3/python/tvm/relay/frontend/onnx.py#L3164
   
   It seems this line does not check `inputs[1].data.numpy()` has more than one 
element. for example of ReduceSum:
   
![图片](https://user-images.githubusercontent.com/2049889/228457151-e7378053-58ae-4680-8862-7026712d96d5.png)
   
   


-- 
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]

Reply via email to