apply constexpr and if constexpr when possible (#102471)
Now that we have full C++17 support, we can use if constexpr in some identified cases.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/102471
Approved by: https://github.com/Skylion007
