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 ...@@ -69,7 +69,7 @@ struct onnx_parser
add_mem_op("Squeeze", &onnx_parser::parse_squeeze); add_mem_op("Squeeze", &onnx_parser::parse_squeeze);
add_mem_op("Unsqueeze", &onnx_parser::parse_unsqueeze); add_mem_op("Unsqueeze", &onnx_parser::parse_unsqueeze);
add_mem_op("Slice", &onnx_parser::parse_slice); 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> template <class F>
...@@ -209,13 +209,13 @@ struct onnx_parser ...@@ -209,13 +209,13 @@ struct onnx_parser
return prog.add_instruction(op, args[0]); return prog.add_instruction(op, args[0]);
} }
// instruction_ref instruction_ref
// parse_concat(const std::string&, attribute_map attributes, std::vector<instruction_ref> args) parse_concat(const std::string&, attribute_map attributes, std::vector<instruction_ref> args)
// { {
// literal axis = parse_value(attributes.at("axis")).at<int>(); std::size_t axis = parse_value(attributes.at("axis")).at<int>();
// op::concat op{axis}; op::concat op{axis};
// return prog.add_instruction(op, std::move(args)); return prog.add_instruction(op, std::move(args));
// } }
instruction_ref instruction_ref
parse_slice(const std::string&, attribute_map attributes, std::vector<instruction_ref> args) 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