Home
last modified time | relevance | path

Searched refs:weightShape (Results 1 – 4 of 4) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/Transforms/
H A DTosaDecomposeConv2D.cpp49 ArrayRef<int64_t> weightShape = weightType.getShape(); in matchAndRewrite() local
50 if (weightShape[1] != 1 || weightShape[2] != 1) in matchAndRewrite()
68 llvm::SmallVector<int64_t, 2> revisedWeightShape{weightShape[0], in matchAndRewrite()
69 weightShape[3]}; in matchAndRewrite()
80 llvm::SmallVector<int64_t, 2> fullyConnectedShape{combined, weightShape[0]}; in matchAndRewrite()
102 inputShape[2], weightShape[0]}; in matchAndRewrite()
H A DTosaDecomposeDepthwise.cpp54 ArrayRef<int64_t> weightShape = weightType.getShape(); in matchAndRewrite() local
55 if (weightShape[0] != 1 || weightShape[1] != 1) { in matchAndRewrite()
73 llvm::SmallVector<int64_t, 2> revisedWeightShape{1, 1, 1, weightShape[2], in matchAndRewrite()
74 weightShape[3]}; in matchAndRewrite()
87 weightShape[3]}; in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/IR/
H A DTosaOps.cpp440 if (weightShape.hasRank()) { in inferReturnTypeComponents()
441 outShape[1] = weightShape.getDimSize(0); in inferReturnTypeComponents()
1036 if (weightShape.hasRank()) { in inferReturnTypeComponents()
1039 weightWidth = weightShape.getDimSize(2); in inferReturnTypeComponents()
1106 if (weightShape.hasRank()) { in inferReturnTypeComponents()
1109 weightWidth = weightShape.getDimSize(2); in inferReturnTypeComponents()
1110 weightDepth = weightShape.getDimSize(3); in inferReturnTypeComponents()
1198 if (weightShape.hasRank()) { in inferReturnTypeComponents()
1200 weightWidth = weightShape.getDimSize(1); in inferReturnTypeComponents()
1277 if (weightShape.hasRank()) { in inferReturnTypeComponents()
[all …]
/llvm-project-15.0.7/mlir/lib/Conversion/TosaToLinalg/
H A DTosaToLinalgNamed.cpp198 auto weightShape = weightTy.getShape(); in matchAndRewrite() local
232 SmallVector<int64_t> newWeightShape{weightShape[1], weightShape[2], in matchAndRewrite()
233 weightShape[3], weightShape[0]}; in matchAndRewrite()
373 auto weightShape = weightTy.getShape(); in matchAndRewrite() local
417 weightShape[2], weightShape[3]}, in matchAndRewrite()
572 auto weightShape = weightTy.getShape(); in matchAndRewrite() local
617 SmallVector<int64_t> newWeightShape{weightShape[1], weightShape[0]}; in matchAndRewrite()