Minor fix to enforce CHECKs in native_function.mlir.test.
Minor fix to enforce CHECKs in native_function.mlir.test.

PiperOrigin-RevId: 356556921

tensorflow/runtime,tensorflow/runtime,tensorflow/runtime,tensorflow/runtime