Replace `tensorflow::Status::OK()` with `tensorflow::{Status, OkStatus}()`.
Replace `tensorflow::Status::OK()` with `tensorflow::{Status, OkStatus}()`.

PiperOrigin-RevId: 478011988

google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research