| 73 | } |
| 74 | |
| 75 | void TF_RegisterOpDefinition(TF_OpDefinitionBuilder* builder, |
| 76 | TF_Status* status) { |
| 77 | auto* cc_builder = reinterpret_cast<OpDefBuilder*>(builder); |
| 78 | TF_SetStatus(status, TF_OK, ""); |
| 79 | ::tensorflow::OpRegistry::Global()->Register( |
| 80 | [cc_builder](::tensorflow::OpRegistrationData* op_reg_data) -> Status { |
| 81 | Status result = cc_builder->Finalize(op_reg_data); |
| 82 | delete cc_builder; |
| 83 | return result; |
| 84 | }); |
| 85 | } |
| 86 | |
| 87 | void TF_OpDefinitionBuilderSetShapeInferenceFunction( |
| 88 | TF_OpDefinitionBuilder* builder, |