Skip to content

GitLab

Commit 1a0abc65 authored by Scott Thornton's avatar Scott Thornton
Browse files

enabled concat onnx parsing

parent 7255bc66
Hide whitespace changes
Inline Side-by-side
Showing with 8 additions and 8 deletions
src/onnx/onnx.cpp
View file @ 1a0abc65
......@@ -69,7 +69,7 @@ struct onnx_parser
add_mem_op("Squeeze", &onnx_parser::parse_squeeze);
add_mem_op("Unsqueeze", &onnx_parser::parse_unsqueeze);
add_mem_op("Slice", &onnx_parser::parse_slice);
// add_mem_op("Concat", &onnx_parser::parse_concat);
add_mem_op("Concat", &onnx_parser::parse_concat);
}
template <class F>
......@@ -209,13 +209,13 @@ struct onnx_parser
return prog.add_instruction(op, args[0]);
}
// instruction_ref
// parse_concat(const std::string&, attribute_map attributes, std::vector<instruction_ref> args)
// {
// literal axis = parse_value(attributes.at("axis")).at<int>();
// op::concat op{axis};
// return prog.add_instruction(op, std::move(args));
// }
instruction_ref
parse_concat(const std::string&, attribute_map attributes, std::vector<instruction_ref> args)
{
std::size_t axis = parse_value(attributes.at("axis")).at<int>();
op::concat op{axis};
return prog.add_instruction(op, std::move(args));
}
instruction_ref
parse_slice(const std::string&, attribute_map attributes, std::vector<instruction_ref> args)
......
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