| 421 | } |
| 422 | |
| 423 | Node* Gather(Graph* g, Node* in0, Node* in1, Node* axis) { |
| 424 | Node* ret; |
| 425 | TF_CHECK_OK(NodeBuilder(g->NewName("n"), "GatherV2") |
| 426 | .Input(in0) |
| 427 | .Input(in1) |
| 428 | .Input(axis) |
| 429 | .Finalize(g, &ret)); |
| 430 | return ret; |
| 431 | } |
| 432 | |
| 433 | Node* GetSessionTensor(Graph* g, Node* in) { |
| 434 | Node* ret; |
no test coverage detected