diff --git a/src/dialect/onnx/onnx_ops.cpp b/src/dialect/onnx/onnx_ops.cpp index 44332b5..cb9407f 100644 --- a/src/dialect/onnx/onnx_ops.cpp +++ b/src/dialect/onnx/onnx_ops.cpp @@ -464,8 +464,8 @@ void ONNXConvNoBiasOp::inferShapes() { if (!getOperand(0).getType().isa() || !getOperand(1).getType().isa()) return; - auto dataTy = getOperand(0)->getType().cast(); - auto weightTy = getOperand(1)->getType().cast(); + auto dataTy = getOperand(0).getType().cast(); + auto weightTy = getOperand(1).getType().cast(); auto dataShape = dataTy.getShape(); auto weightShape = weightTy.getShape();