| 43 | } |
| 44 | |
| 45 | void SampleNBInputs(const NodeDef& pre_node, NodeDef* node) { |
| 46 | if (pre_node.name_ == "API_SAMPLE_NB" || |
| 47 | pre_node.name_ == "API_GATHER_RESULT" || |
| 48 | pre_node.name_ == "API_GET_RNB_NODE" || |
| 49 | pre_node.name_ == "API_GET_NB_NODE" || |
| 50 | pre_node.name_ == "API_GET_NB_FILTER" || |
| 51 | pre_node.name_ == "API_SAMPLE_N_WITH_TYPES") { |
| 52 | node->input_edges_.push_back({pre_node.name_, pre_node.id_, 1}); |
| 53 | } else { |
| 54 | NORMAL_INPUT_GEN(); |
| 55 | } |
| 56 | } |
| 57 | |
| 58 | void GetNBEdgeInputs(const NodeDef& pre_node, NodeDef* node) { |
| 59 | if (pre_node.name_ == "API_SAMPLE_NB" || |
nothing calls this directly
no outgoing calls
no test coverage detected