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
	mlir
	Include the generated interface declarations.
	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. 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, WinogradConv2DFmr fmr)
	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...

Variable Documentation

cols

int64_t cols

Definition at line 195 of file WinogradConv2D.cpp.

Referenced by mlir::presburger::IntegerRelation::isObviouslyEqual(), mlir::presburger::IntMatrix::normalizeRow(), and mlir::presburger::IntegerRelation::removeDuplicateConstraints().

rows

int64_t rows

Definition at line 194 of file WinogradConv2D.cpp.

Referenced by mlir::spirv::CooperativeMatrixType::get(), mlir::presburger::detail::getDual(), and mlir::presburger::Matrix< T >::reserveRows().

scalarFactor

int64_t scalarFactor

Definition at line 196 of file WinogradConv2D.cpp.

table

const float* table

Definition at line 193 of file WinogradConv2D.cpp.

Referenced by getOverloadedDeclaration().