pytorch
b9dc5186 - Add more shape functions for mobilenet (#58932)

Commit
3 years ago
Add more shape functions for mobilenet (#58932) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/58932 This adds all the operators necessary for mobilenet. I kind of wanted to get these landed to unblock ZolotukhinM, but I'm happy to split these up into multiple PRs if it makes reviewing easier. In terms of testing, i'm going to add an automated shape analysis OpInfo test. Test Plan: Imported from OSS Reviewed By: mrshenli Differential Revision: D28727246 Pulled By: eellison fbshipit-source-id: c17f9b7bdf7a43ddf99212b281ae2dd311259374
Author
Parents
Loading