masahi commented on issue #5049: [Relay, TOPI] Refactor Adaptive pool and add 
3d support
URL: https://github.com/apache/incubator-tvm/pull/5049#issuecomment-599003990
 
 
   @kevinthesun Thanks, the diff is not small, but it is due to Relay 
boilerplate that were mostly copy-pasted from 2d version. The only interesting 
bits is cpp topi refactoring.
   
   I also finished adding 3d model support in our torch frontend, buidling on 
this PR. We can now load and run [resnet 
3D](https://github.com/pytorch/vision/blob/master/torchvision/models/video/resnet.py#L286)
 from torchvision. I believe this is the first time TVM runs on 3D workload end 
to end. 
   
   The diff between this PR is here 
https://github.com/masahi/tvm/compare/adaptive-pool-3d...masahi:torch-3d 
   
   cc @jwfromm @optima2005 

----------------------------------------------------------------
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.
 
For queries about this service, please contact Infrastructure at:
[email protected]


With regards,
Apache Git Services

Reply via email to