[aibench] added support for measuring memory on AI Bench for Caffe2 Models (#35036)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/35036
Exposing the helper functions in benchmark_helper.h
Reviewed By: kimishpatel, geof90
Differential Revision: D20528983
fbshipit-source-id: 73231becd93b1e700d37af425bebb628890dec9a
