Fixed AddressSanitizer issue (#2122)
This commit is contained in:
parent
3a1667c35c
commit
135ae12b0d
@ -245,7 +245,8 @@ namespace
|
||||
output->set_shape(output_shape);
|
||||
evaluate(data_tensor, output, lower_bounds, upper_bounds);
|
||||
lower_bounds.at(axis) = upper_bounds.at(axis);
|
||||
upper_bounds.at(axis) += split_lengths[split_pos];
|
||||
if (split_pos < split_lengths.size())
|
||||
upper_bounds.at(axis) += split_lengths[split_pos];
|
||||
}
|
||||
|
||||
return true;
|
||||
|
Loading…
Reference in New Issue
Block a user