Always try to read first in `register_context_if_not_exist` to reduce the chance of concurrent write.
Always try to read first in `register_context_if_not_exist` to reduce the chance of concurrent write.

PiperOrigin-RevId: 343129891

tensorflow/tfx,tensorflow/tfx