Optimize Scale function (#44913)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/44913
Pull Request resolved: https://github.com/pytorch/pytorch/pull/18322
Optimize Scale function
i-am-not-moving-c2-to-c10
Test Plan: buck test mode/dbg caffe2/caffe2/python/operator_test:weighted_sum_test
Reviewed By: BIT-silence
Differential Revision: D14575780
fbshipit-source-id: db333a7964581dcaff6e432ff1d6b517ba1a075f
