code: def _test_smoke(
      self,
      device,
      dtype,
      img_size: int,
      expected_resolution: int,
      model_name: str,
  ):
      model = getattr(vit, f"efficientvit_backbone_{model_name}")()
      model = model.to(device=device, dtype=dtype)
  
      image = torch.randn(
          1, 3, img_size, img_size, device=device, dtype=dtype
      )
  
      out = model(image)
  
      assert "input" in out
      assert out["input"].shape == image.shape
  
      assert "stage_final" in out
      assert out["stage_final"].shape[-2:] == torch.Size(
          [expected_resolution, expected_resolution]
      )
  
file_path: kornia/test/contrib/models/test_efficient_vit.py
imports: from pathlib import Path
  import pytest
  import torch
  from kornia.contrib.models.efficient_vit import EfficientViT, EfficientViTConfig
  from kornia.contrib.models.efficient_vit import backbone
  from kornia.utils._compat import torch_version_lt
label_desc: Use image node_id 1 for calling the TestEfficientViT obj's underlying member method code with example usage: obj._test_smoke(device, dtype, img_size, expected_resolution, model_name) without return types