tests/ops: restore missing x parametrize for test_batchnorm

Accidentally dropped when TestAtan2 was inserted at the same location.

Author: gokulkrishna98

tests/ops/test_ops.py

@@ -653,6 +653,13 @@ async def test_broadcast_shapes(self) -> None:
         await validate_numerical_output(model=model, y=y, x=x)
 
 
+@pytest.mark.parametrize(
+    "x",
+    [
+        torch.rand(2, 3, 8, 8),
+        torch.rand(2, 3, 8, 8, dtype=torch.float16),  # fp16
+    ],
+)
 @pytest.mark.parametrize(
     "dynamic_dims", [tuple(), (0,), (2,), (3,), (0, 2), (0, 3), (0, 2, 3)]
 )