Reland perf fix for nan inf check (#7184)
replace previous usage with logical ops for nan/inf detect with
torch.where
---------
Signed-off-by: Nadav Elyahu <nelyahu@habana.ai>
Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com>
Co-authored-by: Hongwei Chen <33092912+hwchen2017@users.noreply.github.com>
