Minor op cleanup to use ET_KERNEL_CHECK (#1564)

Summary:
Pull Request resolved: https://github.com/pytorch/executorch/pull/1564

Minor cleanup for additional ops to use ET_KERNEL_CHECK_ family macros in preparation for non-fatal kernels.

Example failure log for relu after changes:
```
E 00:00:00.000030 executorch:op_relu.cpp:34] Check failed (tensors_have_same_shape_and_dtype(in, out)):
*** Aborted at 1704827127 (Unix time, try 'date -d 1704827127') ***
...stack trace....
```

Reviewed By: SS-JIA, manuelcandales, kirklandsign

Differential Revision: D52618787

fbshipit-source-id: e3c188c986c1bbb12e05d93b091795f8bc80b18c