code: def test_jit(self, device, dtype):
      B, C, H, W = 2, 1, 32, 32
      patches = torch.ones(B, C, H, W, device=device, dtype=dtype)
      tfeat = TFeat(True).to(patches.device, patches.dtype).eval()
      tfeat_jit = torch.jit.script(
          TFeat(True).to(patches.device, patches.dtype).eval()
      )
      assert_close(tfeat_jit(patches), tfeat(patches))
  
file_path: kornia/test/feature/test_tfeat.py
imports: import pytest
  import torch
  from torch.autograd import gradcheck
  import kornia.testing
  from kornia.feature import TFeat
  from kornia.testing import assert_close
label_desc: Use image node_id 5 for calling the TestTFeat obj's underlying member method code with example usage: obj.test_jit(device, dtype) without return types