Update amax/amin/norm/count_nonzero signatures with int[*]? dim (#83300)

Changes `dim` arg to use `int[*]?` type for the following functions in `native_funcitons.yaml`:
* `amax`
* `amin`
* `norm`
* `frobenius_norm`
* `native_norm`
* `count_nonzero`

Part of #29137

Pull Request resolved: https://github.com/pytorch/pytorch/pull/83300
Approved by: https://github.com/ngimel, https://github.com/albanD, https://github.com/kulinseth

test/test_namedtensor.py

@@ -1190,6 +1190,8 @@ class TestNamedTensor(TestCase):
                 'var_mean',
                 'nanmean',
                 'nansum',
+                'amax',
+                'amin',
             ]
             if op.__name__ in ops_support_dim_none:
                 check_output(op(t, None), [])