| 793 | } |
| 794 | |
| 795 | void TF_AddInputList(TF_OperationDescription* desc, const TF_Output* inputs, |
| 796 | int num_inputs) { |
| 797 | std::vector<NodeBuilder::NodeOut> input_list; |
| 798 | input_list.reserve(num_inputs); |
| 799 | for (int i = 0; i < num_inputs; ++i) { |
| 800 | input_list.emplace_back(&inputs[i].oper->node, inputs[i].index); |
| 801 | } |
| 802 | desc->node_builder.Input(input_list); |
| 803 | } |
| 804 | |
| 805 | void TF_AddControlInput(TF_OperationDescription* desc, TF_Operation* input) { |
| 806 | desc->node_builder.ControlInput(&input->node); |