Skip to content

Commit

Permalink
[NVIDIA] Fix MatMul test (#711)
Browse files Browse the repository at this point in the history
  • Loading branch information
nkogteva authored Aug 29, 2023
1 parent b64d0f4 commit 8582984
Showing 1 changed file with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -201,10 +201,10 @@ class FullyConnectedLayer2MatMulTest : public testing::WithParamInterface<FullyC

std::shared_ptr<ov::Node> matmul1SecondaryInput;
if (secondaryInputType == ngraph::helpers::InputLayerType::PARAMETER) {
matmul1SecondaryInput = std::make_shared<ov::op::v0::Parameter>(ngPrc, ov::Shape(shapeRelatedParams.matmul1_input2.first));
matmul1SecondaryInput = std::make_shared<ov::op::v0::Parameter>(ngPrc, ov::Shape(shapeRelatedParams.matmul2_input2.first));
params.push_back(std::static_pointer_cast<ov::op::v0::Parameter>(matmul1SecondaryInput));
} else {
matmul1SecondaryInput = std::make_shared<ov::op::v0::Constant>(ngPrc, shapeRelatedParams.matmul1_input2.first);
matmul1SecondaryInput = std::make_shared<ov::op::v0::Constant>(ngPrc, shapeRelatedParams.matmul2_input2.first);
}

auto paramOuts = ngraph::helpers::convert2OutputVector(ngraph::helpers::castOps2Nodes<ov::op::v0::Parameter>(params));
Expand Down

0 comments on commit 8582984

Please sign in to comment.