| 468 | } |
| 469 | |
| 470 | Node* Conv2D(Graph* g, Node* in0, Node* in1) { |
| 471 | Node* ret; |
| 472 | TF_CHECK_OK(NodeBuilder(g->NewName("n"), "Conv2D") |
| 473 | .Input(in0) |
| 474 | .Input(in1) |
| 475 | .Attr("T", DT_FLOAT) |
| 476 | .Attr("strides", {1, 1, 1, 1}) |
| 477 | .Attr("padding", "SAME") |
| 478 | .Finalize(g, &ret)); |
| 479 | return ret; |
| 480 | } |
| 481 | |
| 482 | Node* Diag(Graph* g, Node* in, DataType type) { |
| 483 | Node* ret; |
no test coverage detected