Initializes given ApiDef with data in OpDef.
| 304 | |
| 305 | // Initializes given ApiDef with data in OpDef. |
| 306 | void InitApiDefFromOpDef(const OpDef& op_def, ApiDef* api_def) { |
| 307 | api_def->set_graph_op_name(op_def.name()); |
| 308 | api_def->set_visibility(ApiDef::VISIBLE); |
| 309 | |
| 310 | auto* endpoint = api_def->add_endpoint(); |
| 311 | endpoint->set_name(op_def.name()); |
| 312 | |
| 313 | for (const auto& op_in_arg : op_def.input_arg()) { |
| 314 | auto* api_in_arg = api_def->add_in_arg(); |
| 315 | api_in_arg->set_name(op_in_arg.name()); |
| 316 | api_in_arg->set_rename_to(op_in_arg.name()); |
| 317 | api_in_arg->set_description(op_in_arg.description()); |
| 318 | |
| 319 | *api_def->add_arg_order() = op_in_arg.name(); |
| 320 | } |
| 321 | for (const auto& op_out_arg : op_def.output_arg()) { |
| 322 | auto* api_out_arg = api_def->add_out_arg(); |
| 323 | api_out_arg->set_name(op_out_arg.name()); |
| 324 | api_out_arg->set_rename_to(op_out_arg.name()); |
| 325 | api_out_arg->set_description(op_out_arg.description()); |
| 326 | } |
| 327 | for (const auto& op_attr : op_def.attr()) { |
| 328 | auto* api_attr = api_def->add_attr(); |
| 329 | api_attr->set_name(op_attr.name()); |
| 330 | api_attr->set_rename_to(op_attr.name()); |
| 331 | if (op_attr.has_default_value()) { |
| 332 | *api_attr->mutable_default_value() = op_attr.default_value(); |
| 333 | } |
| 334 | api_attr->set_description(op_attr.description()); |
| 335 | } |
| 336 | api_def->set_summary(op_def.summary()); |
| 337 | api_def->set_description(op_def.description()); |
| 338 | } |
| 339 | |
| 340 | // Updates base_arg based on overrides in new_arg. |
| 341 | void MergeArg(ApiDef::Arg* base_arg, const ApiDef::Arg& new_arg) { |
no test coverage detected