Home
last modified time | relevance | path

Searched refs:outputShape (Results 1 – 8 of 8) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/IR/
H A DTosaOps.cpp849 outputShape[axisVal] = 1; in ReduceInferReturnTypes()
976 outputShape.resize(4, -1); in NARY_SHAPE_INFER()
1045 outputShape[3] = ShapedType::isDynamic(outputShape[3]) in inferReturnTypeComponents()
1116 outputShape[4] = in inferReturnTypeComponents()
1117 (outputShape[4] == -1) ? biasShape.getDimSize(0) : outputShape[4]; in inferReturnTypeComponents()
1217 outputShape[3] = ShapedType::isDynamic(outputShape[3]) in inferReturnTypeComponents()
1268 outputShape[0] = ShapedType::isDynamic(outputShape[0]) in inferReturnTypeComponents()
1278 outputShape[3] = ShapedType::isDynamic(outputShape[3]) in inferReturnTypeComponents()
1288 outputShape[3] = ShapedType::isDynamic(outputShape[3]) in inferReturnTypeComponents()
1303 outputShape[1] = outputShape[1] == -1 ? calculateSize : outputShape[1]; in inferReturnTypeComponents()
[all …]
H A DTosaCanonicalizations.cpp311 ArrayRef<int64_t> outputShape = outputType.getShape(); in matchAndRewrite() local
312 if (outputShape[1] != 1 || outputShape[2] != 1) { in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/Transforms/
H A DTosaDecomposeDepthwise.cpp98 auto outputShape = op.getOutput().getType().cast<ShapedType>().getShape(); in matchAndRewrite() local
100 outputShape, in matchAndRewrite()
104 rewriter.getI64ArrayAttr(outputShape)); in matchAndRewrite()
H A DTosaFoldConstantTranspose.cpp59 ArrayRef<int64_t> outputShape = outputType.getShape(); in matchAndRewrite() local
75 dstLinearIndex = dstLinearIndex * outputShape[dim] + dstIndices[dim]; in matchAndRewrite()
H A DTosaDecomposeConv2D.cpp101 llvm::SmallVector<int64_t, 4> outputShape{inputShape[0], inputShape[1], in matchAndRewrite() local
105 rewriter.getI64ArrayAttr(outputShape)); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DTransforms.cpp865 auto outputShape = outputType.getShape(); in returningMatchAndRewrite() local
870 int64_t ohSize = outputShape[1], owSize = outputShape[2]; in returningMatchAndRewrite()
935 auto outputShape = outputType.getShape(); in returningMatchAndRewrite() local
940 int64_t ohSize = outputShape[1], owSize = outputShape[2]; in returningMatchAndRewrite()
H A DConstantFold.cpp160 auto outputShape = outputType.getShape(); in matchAndRewrite() local
201 dstLinearIndex = dstLinearIndex * outputShape[dim] + dstIndices[dim]; in matchAndRewrite()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Tosa/IR/
H A DTosaOpBase.td140 "ArrayAttr":$outpad, "ArrayAttr":$stride, "ArrayAttr":$outputShape),
145 outputShape);