Home
last modified time | relevance | path

Searched refs:reshapedInput (Results 1 – 2 of 2) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/Transforms/
H A DTosaDecomposeConv2D.cpp61 auto reshapedInput = rewriter in matchAndRewrite() local
89 op.getLoc(), fullyConnectedShapeType, reshapedInput, in matchAndRewrite()
96 reshapedInput, reshapedWeight, op.getBias()) in matchAndRewrite()
H A DTosaDecomposeDepthwise.cpp66 auto reshapedInput = rewriter in matchAndRewrite() local
93 .create<tosa::MulOp>(op.getLoc(), mulShapeType, reshapedInput, in matchAndRewrite()