ConvolutedDog commented on issue #18132:
URL: https://github.com/apache/tvm/issues/18132#issuecomment-3065996308
The error occurs because the `expand_shape` parameter you passed to the main
function is `tvm.nd.array([1, 1, 4, 4])`, but TVM's ***Exapnd*** node requires
that the `expand_shape` passed in is the size of the target tensor. Therefore,
`tvm.nd.array([1, 3, 4, 4])` should be passed in here:
```bash
if params:
input_list += params["main"]
input_list[-1] = tvm.nd.array([1, 3, 4, 4])
```
--
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]
