Merge pull request #406 from Xilinx/chaitany.convtranspose_fixing_onnxpad_for_weights

fixing the onnx.pad constant used for padding the weights

Author: chaitanyakamarapu

src/Dialect/ONNX/Transforms/Decompose.cpp

@@ -1549,9 +1549,10 @@ Value decomposeIntoPhasedConvs(PatternRewriter &rewriter, Location loc,
     }
     auto onnxPadsValueConstant =
         getONNXConstOpFromVector(rewriter, loc, weightsPadValue);
-    RankedTensorType scalarTy = RankedTensorType::get({}, elementType);
-    Value onnxPaddingConstantZero = create.onnx.constant(
-        DenseElementsAttr::get(scalarTy, rewriter.getZeroAttr(elementType)));
+    auto weightsElementType = weightsType.getElementType();
+    RankedTensorType scalarTy = RankedTensorType::get({}, weightsElementType);
+    Value onnxPaddingConstantZero = create.onnx.constant(DenseElementsAttr::get(
+        scalarTy, rewriter.getZeroAttr(weightsElementType)));
 
     auto onnxAxisValueConstantNone = create.onnx.none();
     auto wts_shape = weightsType.getShape();