apache / tvm

Open deep learning compiler stack for cpu, gpu and specialized accelerators
https://tvm.apache.org/
Apache License 2.0
11.81k stars 3.48k forks source link

[FRONTEND][PYTORCH ] Support for pixel_unshuffle #17514

Closed srkreddy1238 closed 1 week ago

srkreddy1238 commented 1 week ago

Pixel unshuffle support to pytorch frontend added

tqchen commented 1 week ago

@srkreddy1238 as we are moving towards relax and planning on phasing out relay, would be great to brinig pipelines to relax focused stages