Fix operand type access.

This commit is contained in:
Doru Bercea 2020-01-20 15:46:15 -05:00
parent ab8e2f9a1b
commit 3fe0f2e735
1 changed files with 2 additions and 2 deletions

View File

@ -464,8 +464,8 @@ void ONNXConvNoBiasOp::inferShapes() {
if (!getOperand(0).getType().isa<RankedTensorType>() ||
!getOperand(1).getType().isa<RankedTensorType>())
return;
auto dataTy = getOperand(0)->getType().cast<RankedTensorType>();
auto weightTy = getOperand(1)->getType().cast<RankedTensorType>();
auto dataTy = getOperand(0).getType().cast<RankedTensorType>();
auto weightTy = getOperand(1).getType().cast<RankedTensorType>();
auto dataShape = dataTy.getShape();
auto weightShape = weightTy.getShape();