From 3fe0f2e735785606ef6bad5644f9102a82c97721 Mon Sep 17 00:00:00 2001
From: Doru Bercea <gheorghe-teod.bercea@ibm.com>
Date: Mon, 20 Jan 2020 15:46:15 -0500
Subject: [PATCH] Fix operand type access.

---
 src/dialect/onnx/onnx_ops.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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<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();