Populate operator_input_sizes_ (#68542)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/68542
title
Test Plan: unittest
Reviewed By: iseeyuan
Differential Revision: D32508159
fbshipit-source-id: 0773a725973a493f19a2e9a340365e559dfdf7f8