1 | /* |
---|---|
2 | * SPDX-License-Identifier: Apache-2.0 |
3 | */ |
4 | |
5 | #pragma once |
6 | |
7 | #include <string> |
8 | #include <vector> |
9 | #include "onnx/common/status.h" |
10 | #include "onnx/onnx-operators_pb.h" |
11 | |
12 | namespace ONNX_NAMESPACE { |
13 | |
14 | // Helper function for register nodes in |
15 | // a FunctionProto. Attributes need to be |
16 | // registered separately. |
17 | Common::Status BuildNode( |
18 | const std::string& name, |
19 | const std::string& domain, |
20 | const std::string& doc_string, |
21 | const std::string& op_type, |
22 | std::vector<std::string> const& inputs, |
23 | std::vector<std::string> const& outputs, |
24 | /*OUT*/ NodeProto* node); |
25 | } // namespace ONNX_NAMESPACE |
26 |