| 1423 | } |
| 1424 | |
| 1425 | FunctionDefLibrary FunctionLibraryDefinition::ToProto() const { |
| 1426 | FunctionDefLibrary lib; |
| 1427 | tf_shared_lock l(mu_); |
| 1428 | for (const auto& f : function_defs_) { |
| 1429 | *lib.add_function() = f.second->fdef; |
| 1430 | } |
| 1431 | for (const auto& g : func_grad_) { |
| 1432 | GradientDef* gd = lib.add_gradient(); |
| 1433 | gd->set_function_name(g.first); |
| 1434 | gd->set_gradient_func(g.second); |
| 1435 | } |
| 1436 | return lib; |
| 1437 | } |
| 1438 | |
| 1439 | template <typename T> |
| 1440 | Status FunctionLibraryDefinition::GetAttr(const NodeDef& ndef, |