Address review feedback on device tensor helpers (#20078) #7430

Job	Run time
test-xnnpack / test-backend-linux (xnnpack, operators) / linux-job	14m 40s
test-xnnpack / test-backend-linux (xnnpack, models) / linux-job	20m 18s
test-xnnpack / test-backend-macos	-1s
test-xnnpack / package-golden-artifacts	3m 57s
	38m 54s

Provide feedback