| 168 | } |
| 169 | |
| 170 | void AddOpHelper(TF_Operation* l, TF_Operation* r, TF_Graph* graph, |
| 171 | TF_Status* s, const char* name, TF_Operation** op, |
| 172 | bool check) { |
| 173 | TF_OperationDescription* desc = TF_NewOperation(graph, "AddN", name); |
| 174 | TF_Output add_inputs[2] = {{l, 0}, {r, 0}}; |
| 175 | TF_AddInputList(desc, add_inputs, 2); |
| 176 | *op = TF_FinishOperation(desc, s); |
| 177 | if (check) { |
| 178 | ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s); |
| 179 | ASSERT_NE(*op, nullptr); |
| 180 | } |
| 181 | } |
| 182 | |
| 183 | TF_Operation* Add(TF_Operation* l, TF_Operation* r, TF_Graph* graph, |
| 184 | TF_Status* s, const char* name) { |
no test coverage detected