Spaces:
Running
Running
File size: 989 Bytes
dc2106c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 |
// Copyright (c) ONNX Project Contributors
//
// SPDX-License-Identifier: Apache-2.0
#include "onnx/common/model_helpers.h"
#include "onnx/checker.h"
#include "onnx/defs/schema.h"
#include "onnx/string_utils.h"
namespace ONNX_NAMESPACE {
Common::Status BuildNode(
const std::string& name,
const std::string& domain,
const std::string& doc_string,
const std::string& op_type,
std::vector<std::string> const& inputs,
std::vector<std::string> const& outputs,
NodeProto* node) {
if (node == NULL) {
return Common::Status(Common::CHECKER, Common::INVALID_ARGUMENT, "node_proto should not be nullptr.");
}
node->set_name(name);
node->set_domain(domain);
node->set_doc_string(doc_string);
node->set_op_type(op_type);
for (auto& input : inputs) {
node->add_input(input);
}
for (auto& output : outputs) {
node->add_output(output);
}
return Common::Status::OK();
}
} // namespace ONNX_NAMESPACE
|