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
12namespace ONNX_NAMESPACE {
13
14// Helper function for register nodes in
15// a FunctionProto. Attributes need to be
16// registered separately.
17Common::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