code: def test_jit(self, device, dtype):
      B, C, H, W = 2, 3, 4, 4
      img = torch.ones(B, C, H, W, device=device, dtype=dtype)
      op = kornia.color.rgb_to_yuv
      op_jit = torch.jit.script(op)
      self.assert_close(op(img), op_jit(img))
  
imports: import pytest
  import torch
  from torch.autograd import gradcheck
  import kornia
  from kornia.testing import BaseTester
label_desc: Use image node_id 6 for calling the TestRgbToYuv obj's underlying member method code with example usage: obj.test_jit(device, dtype) without return types