Commit 99432ec3 authored by Khalique's avatar Khalique
Browse files

fix for zero in shape lens

parent 7d72a081
...@@ -166,7 +166,7 @@ struct onnx_parser ...@@ -166,7 +166,7 @@ struct onnx_parser
if(s0->size() > s1->size()) if(s0->size() > s1->size())
std::swap(s0, s1); std::swap(s0, s1);
std::vector<std::size_t> output_lens(s1->size()); std::vector<std::size_t> output_lens(*s1);
auto offset = s1->size() - s0->size(); auto offset = s1->size() - s0->size();
std::transform(s0->begin(), std::transform(s0->begin(),
s0->end(), s0->end(),
......
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