| 1728 | } |
| 1729 | |
| 1730 | TF_Operation* MatMul(TF_Graph* graph, TF_Status* s, TF_Operation* l, |
| 1731 | TF_Operation* r, const char* name, |
| 1732 | bool transpose_a = false, bool transpose_b = false) { |
| 1733 | TF_OperationDescription* desc = TF_NewOperation(graph, "MatMul", name); |
| 1734 | if (transpose_a) { |
| 1735 | TF_SetAttrBool(desc, "transpose_a", 1); |
| 1736 | } |
| 1737 | if (transpose_b) { |
| 1738 | TF_SetAttrBool(desc, "transpose_b", 1); |
| 1739 | } |
| 1740 | TF_AddInput(desc, {l, 0}); |
| 1741 | TF_AddInput(desc, {r, 0}); |
| 1742 | TF_Operation* op = TF_FinishOperation(desc, s); |
| 1743 | EXPECT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s); |
| 1744 | return op; |
| 1745 | } |
| 1746 | |
| 1747 | TF_Operation* OnesLike(TF_Graph* graph, TF_Status* s, TF_Operation* in, |
| 1748 | const char* name) { |
nothing calls this directly
no test coverage detected