IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[CPU] [Test] Eye testcase does not populate all parameters (#11869)

Browse Source
This commit is contained in:
Luo Cheng 2022-06-14 09:32:56 +08:00 committed by GitHub
parent 151d77062f
commit cae0c924b6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/tests/functional/plugin/cpu/single_layer_tests/eye.cpp
View File

@ -113,7 +113,10 @@ protected:
if (i == 3) { // batch shape
tensor = ov::Tensor(funcInput.get_element_type(), targetInputStaticShapes[i]);
int *batchShapePtr = tensor.data<int>();
for (size_t j = 0; j < targetInputStaticShapes[i].size(); j++) {
// Spec: batch_shape - 1D tensor with non-negative values of type T_NUM defines leading batch dimensions of output shape
EXPECT_EQ(targetInputStaticShapes[i].size(), 1);
EXPECT_EQ(targetInputStaticShapes[i][0], outBatchShape.size());
for (size_t j = 0; j < targetInputStaticShapes[i][0]; j++) {
batchShapePtr[j] = outBatchShape[j];
}
} else {