Searched refs:dstStaticShape (Results 1 – 2 of 2) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorInferTypeOpInterfaceImpl.cpp | 64 ArrayRef<int64_t> dstStaticShape, ArrayRef<AffineMap> reassociation) { in getCollapsedOutputShapeFromInputShape() argument
66 llvm::seq<int64_t>(0, dstStaticShape.size()), [&](int64_t dim) { in getCollapsedOutputShapeFromInputShape()
76 ArrayRef<int64_t> dstStaticShape, ArrayRef<AffineMap> reassociation, in getExpandedOutputDimFromInputShape() argument
78 if (!ShapedType::isDynamic(dstStaticShape[dimIndex])) { in getExpandedOutputDimFromInputShape()
79 return builder.getI64IntegerAttr(dstStaticShape[dimIndex]); in getExpandedOutputDimFromInputShape()
114 ArrayRef<int64_t> dstStaticShape, ArrayRef<AffineMap> reassociation) { in getExpandedOutputShapeFromInputShape() argument
118 llvm::seq<int64_t>(0, dstStaticShape.size()), [&](int64_t dim) { in getExpandedOutputShapeFromInputShape()
127 ArrayRef<int64_t> dstStaticShape, in getReshapeOutputShapeFromInputShape() argument
129 return dstStaticShape.size() > in getReshapeOutputShapeFromInputShape()
132 builder, loc, src, dstStaticShape, reassocation) in getReshapeOutputShapeFromInputShape()
[all …]
|
| /llvm-project-15.0.7/mlir/lib/Conversion/MemRefToLLVM/ |
| H A D | MemRefToLLVM.cpp | 1599 ArrayRef<int64_t> dstStaticShape = dstType.getShape(); in matchAndRewrite() local
1604 srcStaticShape, srcDesc, dstStaticShape); in matchAndRewrite()
|