colin
/
onnx-mlir
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/dialect/onnx/onnx_ops.cpp
View File

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