code: def test_smoke(self, device, dtype, img_size):
      model = TinyViT(img_size=img_size).to(device=device, dtype=dtype)
      inpt = torch.randn(
          1, 3, img_size, img_size, device=device, dtype=dtype
      )
  
      out = model(inpt)
      assert isinstance(out, Tensor)
  
file_path: kornia/test/contrib/models/test_tiny_vit.py
imports: import pytest
  import torch
  from kornia.contrib.models.tiny_vit import TinyViT
  from kornia.core import Tensor
  from kornia.testing import BaseTester
label_desc: Use image node_id 1 for calling the TestTinyViT obj's underlying member method code with example usage: obj.test_smoke(device, dtype, img_size) without return types