code: def test_device_arg(self):
      # Test that no device gets propagated when arg is passed in
      def set_device(x, device_name: torch.device):
          return x.to(device=device_name)
  
      for in_device in self.device_types:
          self.assert_device_equal(set_device, [in_device, None], None)
  
imports: from itertools import product
  import unittest
  import torch
  from torch.testing._internal.common_utils import TEST_CUDA
  from torch.testing._internal.jit_utils import JitTestCase
  from torch.jit._passes._property_propagation import apply_input_props_using_example
label_desc: Use image node_id 9 for calling the TestDeviceAnalysis obj's underlying member method code with example usage: obj.test_device_arg() and returns: x