@@ -1790,8 +1790,8 @@ class ConvertAtenOuterOp : public OpConversionPattern<AtenOuterOp> {
1790
1790
Type newResultType = getTypeConverter ()->convertType (op.getType ());
1791
1791
1792
1792
// Create a zero-initialized tensor with shape [lhsDim, rhsDim]
1793
- Value zeroTensor = createZeroInitTensor (
1794
- rewriter, loc, ValueRange{lhsDim, rhsDim}, elementType);
1793
+ Value initTensor = createInitTensor (
1794
+ rewriter, loc, ValueRange{lhsDim, rhsDim}, elementType, NULL );
1795
1795
1796
1796
// Set up affine indexing maps:
1797
1797
// We create a 2D loop iteration space. For the lhs, we use the first index
@@ -1811,9 +1811,9 @@ class ConvertAtenOuterOp : public OpConversionPattern<AtenOuterOp> {
1811
1811
Value outerProd =
1812
1812
rewriter
1813
1813
.create <linalg::GenericOp>(
1814
- loc, zeroTensor .getType (),
1814
+ loc, initTensor .getType (),
1815
1815
/* inputs=*/ ValueRange{lhsDim, rhsDim},
1816
- /* outputs=*/ zeroTensor ,
1816
+ /* outputs=*/ initTensor ,
1817
1817
/* indexingMaps=*/
1818
1818
SmallVector<AffineMap, 3 >{mapLhs, mapRhs, mapOut},
1819
1819
/* iteratortType=*/ iteratorTypes,
0 commit comments