WinogradConv2D.cpp File Reference

#include "mlir/Dialect/Affine/IR/AffineOps.h"
#include "mlir/Dialect/Arith/IR/Arith.h"
#include "mlir/Dialect/Linalg/IR/Linalg.h"
#include "mlir/Dialect/Linalg/Utils/Utils.h"
#include "mlir/Dialect/Tensor/IR/Tensor.h"
#include "mlir/Dialect/Utils/StaticValueUtils.h"
#include "llvm/Support/MathExtras.h"

Go to the source code of this file.

Namespaces
namespace	mlir
	Include the generated interface declarations.
namespace	mlir::linalg

Functions
FailureOr< Operation * >	mlir::linalg::winogradConv2D (RewriterBase &rewriter, linalg::Conv2DNhwcFhwcOp op, linalg::WinogradConv2DFmr fmr)
FailureOr< Operation * >	mlir::linalg::decomposeWinogradFilterTransformOp (RewriterBase &rewriter, linalg::WinogradFilterTransformOp op)
	Rewrite linalg.winograd_filter_transform.
FailureOr< Operation * >	mlir::linalg::decomposeWinogradInputTransformOp (RewriterBase &rewriter, linalg::WinogradInputTransformOp op)
	Rewrite linalg.winograd_input_transform.
FailureOr< Operation * >	mlir::linalg::decomposeWinogradOutputTransformOp (RewriterBase &rewriter, linalg::WinogradOutputTransformOp op)
	Rewrite linalg.winograd_output_transform.
void	mlir::linalg::populateWinogradConv2DPatterns (RewritePatternSet &patterns, WinogradConv2DFmr fmr)
	Patterns to apply Winograd Conv2D algorithm F(m x m, r x r).
void	mlir::linalg::populateDecomposeWinogradOpsPatterns (RewritePatternSet &patterns)
	Patterns to decompose Winograd operators.