Fix reference error.

This commit is contained in:
Doru Bercea 2020-01-22 16:09:19 -05:00
parent 0bc07ef661
commit 1784ec2314
1 changed files with 3 additions and 3 deletions

View File

@ -328,8 +328,8 @@ void ONNXMatMulOp::inferShapes() {
!getOperand(1).getType().isa<RankedTensorType>())
return;
auto lhsTy = getOperand(0)->getType().cast<RankedTensorType>();
auto rhsTy = getOperand(1)->getType().cast<RankedTensorType>();
auto lhsTy = getOperand(0).getType().cast<RankedTensorType>();
auto rhsTy = getOperand(1).getType().cast<RankedTensorType>();
SmallVector<int64_t, 2> dims;
auto lhsShape = lhsTy.getShape();
@ -413,7 +413,7 @@ void ONNXMatMulOp::inferShapes() {
dims.emplace_back(rhsShape[1]);
}
getResult()->setType(RankedTensorType::get(dims, lhsTy.getElementType()));
getResult().setType(RankedTensorType::get(dims, lhsTy.getElementType()));
}
//===----------------------------------------------------------------------===//