code: def test_mobilenet(self):
      in_cpu = torch.randn(1, 3, 224, 224, device=self.cpu)
      in_example = in_cpu
  
      expected_device = self.cpu
      m = torch.jit.script(models.mobilenet_v3_small())
      m.eval()
      graph = torch.jit.freeze(m).graph
      # torch._C._jit_pass_erase_shape_information(graph)
      apply_input_props_using_example(graph, in_example)
      torch._C._jit_pass_propagate_shapes_on_graph(graph)
      torch._C._jit_pass_propagate_device(graph)
  
      actual_device = self.node_output_device(graph)
  
      if expected_device is None or actual_device is None:
          self.assertEqual(actual_device, expected_device)
      else:
          self.assertEqual(
              actual_device.type,
              expected_device.type,
              "Failed Verification",
          )
  
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 6 for calling the TestDeviceAnalysis obj's underlying member method code with example usage: obj.test_mobilenet() without return types