Reduce input sets for tests to speed them up. (#23692)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/23692
Before tests took ~40s to finish, with this change it's ~2s.
Test Plan: Imported from OSS
Differential Revision: D16611479
Pulled By: ZolotukhinM
fbshipit-source-id: 391235483029d2ab860fcc4597ce84f4964025f1
Author
Mikhail Zolotukhin
