| 29 | typedef FunctionDefHelper FDH; |
| 30 | |
| 31 | GraphDef GDef(gtl::ArraySlice<NodeDef> nodes, |
| 32 | gtl::ArraySlice<FunctionDef> funcs) { |
| 33 | GraphDef g; |
| 34 | VersionDef* versions = g.mutable_versions(); |
| 35 | versions->set_producer(TF_GRAPH_DEF_VERSION); |
| 36 | versions->set_min_consumer(TF_GRAPH_DEF_VERSION_MIN_CONSUMER); |
| 37 | for (const auto& n : nodes) { |
| 38 | *(g.add_node()) = n; |
| 39 | } |
| 40 | auto lib = g.mutable_library(); |
| 41 | for (const auto& f : funcs) { |
| 42 | *(lib->add_function()) = f; |
| 43 | } |
| 44 | return g; |
| 45 | } |
| 46 | |
| 47 | // Helper to construct a NodeDef. |
| 48 | NodeDef NDef(StringPiece name, StringPiece op, gtl::ArraySlice<string> inputs, |
no test coverage detected