Commit 2b697563 authored by Paul's avatar Paul
Browse files

Formatting

parent ea166055
...@@ -24,7 +24,8 @@ struct onnx_parser ...@@ -24,7 +24,8 @@ struct onnx_parser
{ {
using attribute_map = std::unordered_map<std::string, onnx::AttributeProto>; using attribute_map = std::unordered_map<std::string, onnx::AttributeProto>;
using node_map = std::unordered_map<std::string, onnx::NodeProto>; using node_map = std::unordered_map<std::string, onnx::NodeProto>;
using op_func = std::function<std::vector<instruction_ref>(attribute_map, std::vector<instruction_ref>)>; using op_func =
std::function<std::vector<instruction_ref>(attribute_map, std::vector<instruction_ref>)>;
node_map nodes; node_map nodes;
std::unordered_map<std::string, instruction_ref> instructions; std::unordered_map<std::string, instruction_ref> instructions;
program prog = program(); program prog = program();
...@@ -719,9 +720,13 @@ struct onnx_parser ...@@ -719,9 +720,13 @@ struct onnx_parser
{ {
result = ops[node.op_type()](get_attributes(node), args); result = ops[node.op_type()](get_attributes(node), args);
} }
std::transform(node.output().begin(), node.output().end(), result.begin(), std::inserter(instructions, instructions.end()), [](auto&& onnx_out, auto&& parse_out) { std::transform(node.output().begin(),
return std::make_pair(onnx_out, parse_out); node.output().end(),
}); result.begin(),
std::inserter(instructions, instructions.end()),
[](auto&& onnx_out, auto&& parse_out) {
return std::make_pair(onnx_out, parse_out);
});
} }
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment