Increase tensor size for mem check tests (#70603)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/70226
Pull Request resolved: https://github.com/pytorch/pytorch/pull/70603
Reviewed By: mruberry
Differential Revision: D33410439
Pulled By: janeyx99
fbshipit-source-id: e94615ece6d0fdf230de5297118678b70f34a18c