| 91 | } |
| 92 | |
| 93 | Node* FromNodeDef(const string& name, const string& node_type, |
| 94 | int num_inputs) { |
| 95 | auto builder = NodeDefBuilder(name, node_type); |
| 96 | for (int i = 0; i < num_inputs; ++i) { |
| 97 | builder = builder.Input(strings::StrCat("node_", i), i, DT_FLOAT); |
| 98 | } |
| 99 | |
| 100 | NodeDef node_def; |
| 101 | TF_CHECK_OK(builder.Finalize(&node_def)); |
| 102 | |
| 103 | Status s; |
| 104 | Node* node = graph_.AddNode(node_def, &s); |
| 105 | TF_CHECK_OK(s); |
| 106 | return node; |
| 107 | } |
| 108 | |
| 109 | void FromGraphDef(const string& gdef_ascii) { |
| 110 | GraphDef gdef; |
nothing calls this directly
no test coverage detected