| 157 | } |
| 158 | |
| 159 | Node* Matmul(Graph* g, Node* in0, Node* in1, bool transpose_a, |
| 160 | bool transpose_b) { |
| 161 | Node* ret; |
| 162 | TF_CHECK_OK(NodeBuilder(g->NewName("n"), "MatMul") |
| 163 | .Input(in0) |
| 164 | .Input(in1) |
| 165 | .Attr("transpose_a", transpose_a) |
| 166 | .Attr("transpose_b", transpose_b) |
| 167 | .Finalize(g, &ret)); |
| 168 | return ret; |
| 169 | } |
| 170 | |
| 171 | Node* BatchMatmul(Graph* g, Node* in0, Node* in1, bool adj_x, bool adj_y) { |
| 172 | Node* ret; |
no test coverage detected