Searched refs:inputPadding (Results 1 – 1 of 1) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Tosa/Transforms/ |
| H A D | TosaDecomposeTransposeConv.cpp | 255 llvm::SmallVector<int32_t, 8> inputPadding = {0, 0, 0, 0, 0, 0, 0, 0}; in matchAndRewrite() local
256 inputPadding[2] += restridedWeightTy.getDimSize(1) - 1; in matchAndRewrite()
257 inputPadding[3] += restridedWeightTy.getDimSize(1) - 1; in matchAndRewrite()
258 inputPadding[4] += restridedWeightTy.getDimSize(2) - 1; in matchAndRewrite()
259 inputPadding[5] += restridedWeightTy.getDimSize(2) - 1; in matchAndRewrite()
262 RankedTensorType::get({4, 2}, rewriter.getI32Type()), inputPadding); in matchAndRewrite()
|