pytorch

31137a63 - Changed flop formulas for flop counter to take in shapes directly (#96565)

Changed flop formulas for flop counter to take in shapes directly (#96565) Pull Request resolved: https://github.com/pytorch/pytorch/pull/96565 Approved by: https://github.com/zdevito