Fixed AddressSanitizer issue (#2122)

This commit is contained in:
Ilya Churaev 2020-09-09 14:31:12 +03:00 committed by GitHub
parent 3a1667c35c
commit 135ae12b0d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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;