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

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

View File

@ -113,7 +113,10 @@ protected:
if (i == 3) { // batch shape if (i == 3) { // batch shape
tensor = ov::Tensor(funcInput.get_element_type(), targetInputStaticShapes[i]); tensor = ov::Tensor(funcInput.get_element_type(), targetInputStaticShapes[i]);
int *batchShapePtr = tensor.data<int>(); 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]; batchShapePtr[j] = outBatchShape[j];
} }
} else { } else {