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

Job	Run time
gather-models	9s
test-models-linux (cmake, vit, xnnpack-quantization-delegation, linux.2xlarge, 90) / linux-job	10m 17s
test-models-linux (cmake, mv3, portable, linux.2xlarge, 90) / linux-job	9m 22s
test-models-linux (buck2, mv3, portable, linux.2xlarge, 90) / linux-job	9m 40s
test-models-linux (cmake, vit, portable, linux.2xlarge, 90) / linux-job	16m 54s
test-models-linux (buck2, mv3, xnnpack-quantization-delegation, linux.2xlarge, 90) / linux-job	10m 25s
test-models-linux (cmake, mv3, xnnpack-quantization-delegation, linux.2xlarge, 90) / linux-job	9m 59s
	1h 6m 46s

Provide feedback