Update deprecated autocast functions #3603

Job	Run time
linux (linux.12xlarge, cpu) / linux-job	4m 4s
linux (linux.g5.4xlarge.nvidia.gpu, cuda, 11.8) / linux-job	8m 56s
macos (macos-m1-stable) / macos-job	3m 18s
windows (windows.4xlarge, cpu) / windows-job	13m 25s
windows (windows.g5.4xlarge.nvidia.gpu, cuda, 11.8) / windows-job	16m 23s
	46m 6s

Provide feedback