Added missing change

Author: amemov

lib/Conversion/TorchToLinalg/Linear.cpp

@@ -1704,8 +1704,8 @@ class ConvertAtenOuterOp : public OpConversionPattern<AtenOuterOp> {
     Type newResultType = getTypeConverter()->convertType(op.getType());
 
     // Create a zero-initialized tensor with shape [lhsDim, rhsDim]
-    Value zeroTensor = createZeroInitTensor(
-        rewriter, loc, ValueRange{lhsDim, rhsDim}, elementType);
+    Value initTensor = createInitTensor(
+        rewriter, loc, ValueRange{lhsDim, rhsDim}, elementType, NULL);
 
     // Set up affine indexing maps:
     // We create a 2D loop iteration space. For the lhs, we use the first index
@@ -1725,9 +1725,9 @@ class ConvertAtenOuterOp : public OpConversionPattern<AtenOuterOp> {
     Value outerProd =
         rewriter
             .create<linalg::GenericOp>(
-                loc, zeroTensor.getType(),
+                loc, initTensor.getType(),
                 /*inputs=*/ValueRange{lhsDim, rhsDim},
-                /*outputs=*/zeroTensor,
+                /*outputs=*/initTensor,
                 /*indexingMaps=*/
                 SmallVector<AffineMap, 3>{mapLhs, mapRhs, mapOut},
                 /*iteratortType=*/iteratorTypes,