code: def test_jit(self, device, dtype):
      data = torch.ones(2, 3, 1, 1, device=device, dtype=dtype)
      mean = torch.tensor(
          [0.5, 1.0, 2.0], device=device, dtype=dtype
      ).repeat(2, 1)
      std = torch.tensor(
          [2.0, 2.0, 2.0], device=device, dtype=dtype
      ).repeat(2, 1)
      inputs = (data, mean, std)
  
      op = kornia.enhance.normalize
      op_script = torch.jit.script(op)
  
      self.assert_close(op(*inputs), op_script(*inputs))
  
imports: import pytest
  import torch
  from torch.autograd import gradcheck
  import kornia
  import kornia.testing
  from kornia.testing import BaseTester
label_desc: Use image node_id 7 for calling the TestNormalize obj's underlying member method code with example usage: obj.test_jit(device, dtype) without return types