| 432 | } |
| 433 | |
| 434 | std::vector<std::pair<string, string>> GetGradDefs( |
| 435 | const tensorflow::GraphDef& graph_def) { |
| 436 | std::vector<std::pair<string, string>> grads; |
| 437 | for (const tensorflow::GradientDef& grad : graph_def.library().gradient()) { |
| 438 | grads.emplace_back(grad.function_name(), grad.gradient_func()); |
| 439 | } |
| 440 | std::sort(grads.begin(), grads.end()); |
| 441 | return grads; |
| 442 | } |
| 443 | |
| 444 | std::vector<string> GetFuncNames(const tensorflow::GraphDef& graph_def) { |
| 445 | std::vector<string> names; |
no test coverage detected