TestJit.test_import_method.expect

def forward(self,
    x: Tensor,
    y: Tensor) -> Tensor:
  return torch.add(torch.mul(x, 2), y)