Go to the source code of this file.
| Functions |
|
| FailureOr< Operation * > |
mlir::linalg::winogradConv2D (RewriterBase &rewriter, linalg::Conv2DNhwcFhwcOp op, int64_t m, int64_t r) |
|
Convert linalg.conv_2d_nhwc_fhwc to Winograd Conv2D algorithm F(m x m, r x r). More... |
|
|
| FailureOr< Operation * > |
mlir::linalg::decomposeWinogradFilterTransformOp (RewriterBase &rewriter, linalg::WinogradFilterTransformOp op) |
|
Rewrite linalg.winograd_filter_transform. More... |
|
|
| FailureOr< Operation * > |
mlir::linalg::decomposeWinogradInputTransformOp (RewriterBase &rewriter, linalg::WinogradInputTransformOp op) |
|
Rewrite linalg.winograd_input_transform. More... |
|
|
| FailureOr< Operation * > |
mlir::linalg::decomposeWinogradOutputTransformOp (RewriterBase &rewriter, linalg::WinogradOutputTransformOp op) |
|
Rewrite linalg.winograd_output_transform. More... |
|
|
| void |
mlir::linalg::populateWinogradConv2DPatterns (RewritePatternSet &patterns, int64_t m, int64_t r) |
|
Patterns to apply Winograd Conv2D algorithm F(m x m, r x r). More... |
|
|
| void |
mlir::linalg::populateDecomposeWinogradOpsPatterns (RewritePatternSet &patterns) |
|
Patterns to decompose Winograd operators. More... |
|
|
◆ cols
◆ rows
◆ scalarFactor
◆ table