potok/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2026-07-21 19:33:03 +04:00
Code Packages Releases Wiki Activity

bound einsum subscript index before reading input shape

Browse Source
This commit is contained in:
Uwez Khan
2026-07-01 15:14:43 +05:30
parent 5b83b0c440
commit b4164a4bc4
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/dnn/src/layers/einsum_layer.cpp
+3 -2
View File
@@ -953,6 +953,8 @@ void LayerEinsumImpl::processEquation(const std::vector<MatShape>& inputs)
CV_CheckNE(letterIdx, -1, CV_CheckNE(letterIdx, -1,
"The only permissible subscript labels are lowercase letters (a-z) and uppercase letters (A-Z)."); "The only permissible subscript labels are lowercase letters (a-z) and uppercase letters (A-Z).");
CV_CheckLT(dim_count, rank,
"The Einsum subscripts string has an excessive number of subscript labels compared to the rank of the input.");
int dimValue = shape[dim_count]; int dimValue = shape[dim_count];
// The subscript label was not found in the global subscript label array // The subscript label was not found in the global subscript label array
@@ -980,8 +982,7 @@ void LayerEinsumImpl::processEquation(const std::vector<MatShape>& inputs)
++letter2count[letterIdx]; ++letter2count[letterIdx];
currTokenIndices.push_back(letter2index[letterIdx]); currTokenIndices.push_back(letter2index[letterIdx]);
CV_CheckLE(++dim_count, rank, ++dim_count;
"The Einsum subscripts string has an excessive number of subscript labels compared to the rank of the input.");
} }
} }