From de77758faf9cbc64be302cbed74e4870c67772f1 Mon Sep 17 00:00:00 2001
From: Doru Bercea <gheorghe-teod.bercea@ibm.com>
Date: Wed, 22 Jan 2020 10:10:06 -0500
Subject: [PATCH] Fix kernel dimensions.

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

diff --git a/src/dialect/onnx/onnx_ops.cpp b/src/dialect/onnx/onnx_ops.cpp
index e597c9a..687a431 100644
--- a/src/dialect/onnx/onnx_ops.cpp
+++ b/src/dialect/onnx/onnx_ops.cpp
@@ -509,15 +509,15 @@ void ONNXConvNoBiasOp::inferShapes() {
   // argument.
   SmallVector<int64_t, 2> kernelDims;
   if (auto kernelShape = getAttrOfType<ArrayAttr>(
-      ONNXConvOp::getKernelShapeAttrName())) {
+          ONNXConvOp::getKernelShapeAttrName())) {
     if (kernelShape.getValue().size() != nDims)
       emitError("kernel_shape length incompatible with spatial dimensions.");
     for (int i = 0; i < nDims; ++i)
-      kernelDims[i] =
-          (kernelShape.getValue()[i]).cast<IntegerAttr>().getInt();
+      kernelDims.emplace_back(
+          (kernelShape.getValue()[i]).cast<IntegerAttr>().getInt());
   } else {
     for (int i = 0; i < nDims; ++i)
-      kernelDims[i] = weightShape[i + 2];
+      kernelDims.emplace_back(weightShape[i + 2]);
   }
 
   // Subtract kernel dimensions from input data dimensions.