onnx/common/model_helpers.h

0001 // Copyright (c) ONNX Project Contributors
0002 //
0003 // SPDX-License-Identifier: Apache-2.0
0004
0005 #pragma once
0006
0007 #include <string>
0008 #include <vector>
0009
0010 #include "onnx/common/status.h"
0011 #include "onnx/onnx-operators_pb.h"
0012
0013 namespace ONNX_NAMESPACE {
0014
0015 // Helper function for register nodes in
0016 // a FunctionProto. Attributes need to be
0017 // registered separately.
0018 Common::Status BuildNode(
0019     const std::string& name,
0020     const std::string& domain,
0021     const std::string& doc_string,
0022     const std::string& op_type,
0023     std::vector<std::string> const& inputs,
0024     std::vector<std::string> const& outputs,
0025     /*OUT*/ NodeProto* node);
0026 } // namespace ONNX_NAMESPACE