| 50 | } |
| 51 | |
| 52 | void OpRegistry::Register(const OpRegistrationDataFactory& op_data_factory) { |
| 53 | mutex_lock lock(mu_); |
| 54 | if (initialized_) { |
| 55 | TF_QCHECK_OK(RegisterAlreadyLocked(op_data_factory)); |
| 56 | } else { |
| 57 | deferred_.push_back(op_data_factory); |
| 58 | } |
| 59 | } |
| 60 | |
| 61 | Status OpRegistry::LookUp(const string& op_type_name, |
| 62 | const OpRegistrationData** op_reg_data) const { |