| 504 | } |
| 505 | |
| 506 | NodeID GraphBuilder::add_elementwise_node( |
| 507 | Graph &g, NodeParams params, NodeIdxPair input0, NodeIdxPair input1, EltwiseOperation operation) |
| 508 | { |
| 509 | check_nodeidx_pair(input0, g); |
| 510 | check_nodeidx_pair(input1, g); |
| 511 | |
| 512 | NodeID nid = g.add_node<EltwiseLayerNode>(descriptors::EltwiseLayerDescriptor{operation}); |
| 513 | |
| 514 | g.add_connection(input0.node_id, input0.index, nid, 0); |
| 515 | g.add_connection(input1.node_id, input1.index, nid, 1); |
| 516 | |
| 517 | set_node_params(g, nid, params); |
| 518 | |
| 519 | return nid; |
| 520 | } |
| 521 | |
| 522 | NodeID GraphBuilder::add_flatten_node(Graph &g, NodeParams params, NodeIdxPair input) |
| 523 | { |
nothing calls this directly
no test coverage detected