benchmark

475e123a - tweak shapes (#2372)

tweak shapes (#2372) Summary: Pull Request resolved: https://github.com/pytorch/benchmark/pull/2372 Reviewed By: manman-ren, jianyuh Differential Revision: D59535933 Pulled By: bertmaher fbshipit-source-id: b361e66e22f2824c542446b83a75061e02e7cbcc