pytorch

03992a6f - Make the data types of output and input consistenst for batchnorm (#86784)

Make the data types of output and input consistenst for batchnorm (#86784) * make the data types of output and input consistenst for batchnorm * Apply suggestions from code review Co-authored-by: Nikita Shulga <nikita.shulga@gmail.com>