Unverified Commit 52203fa4 authored by Attila Dusnoki's avatar Attila Dusnoki Committed by GitHub
Browse files

Fix wrong size check when axes not present for slice (#2270)

parent 685168f5
...@@ -137,7 +137,7 @@ struct parse_slice : op_parser<parse_slice> ...@@ -137,7 +137,7 @@ struct parse_slice : op_parser<parse_slice>
sd.always_insert(args.at(0)); sd.always_insert(args.at(0));
// If axes arg is not given, the default is all of them. // If axes arg is not given, the default is all of them.
if(sd.op.axes.empty() and sd.op_args.size() < 3) if(sd.op.axes.empty() and sd.op_args.size() <= 3)
{ {
std::vector<int64_t> axes(args[0]->get_shape().ndim()); std::vector<int64_t> axes(args[0]->get_shape().ndim());
std::iota(axes.begin(), axes.end(), int64_t{0}); std::iota(axes.begin(), axes.end(), int64_t{0});
......
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