mlir::linalg::LinalgGenericToLinalgCustomBatchNorm2DNchw Struct Reference

y = (x - E[x]) / sqrt(Var[x] + eps) * gamma + beta More...

#include <mlir/Dialect/Linalg/Transforms/CustomSpecializeBatchNorm.h>

Inheritance diagram for mlir::linalg::LinalgGenericToLinalgCustomBatchNorm2DNchw:

Public Member Functions
bool	is4dIdentityMap (Attribute attr) const
bool	is3dIdentityMap (Attribute attr) const
bool	checkIndexingMaps (ArrayAttr array_attr) const
bool	checkIteratorTypes (ArrayAttr array_attr) const
template<typename ElemOp>
bool	isElementwiseRegion (Region &region) const
template<typename ElemOp>
bool	checkGenericOp (GenericOp generic_op) const
bool	isTypeNCHW (Type type, Type element_type, int64_t n, int64_t c, int64_t h, int64_t w) const
bool	isTypeC11 (Type type, Type element_type, int64_t c) const
bool	isTypeC (Type type, Type element_type, int64_t c) const
bool	checkExpandShape (tensor::ExpandShapeOp expand_shape_op, int64_t c) const
LogicalResult	matchAndRewrite (GenericOp generic_op, PatternRewriter &rewriter) const override

Detailed Description

y = (x - E[x]) / sqrt(Var[x] + eps) * gamma + beta

Member Function Documentation

checkExpandShape()

bool mlir::linalg::LinalgGenericToLinalgCustomBatchNorm2DNchw::checkExpandShape	(	tensor::ExpandShapeOp	expand_shape_op,
		int64_t	c ) const

checkGenericOp()

template<typename ElemOp>

bool mlir::linalg::LinalgGenericToLinalgCustomBatchNorm2DNchw::checkGenericOp ( GenericOp generic_op ) const

inline

checkIndexingMaps()

bool mlir::linalg::LinalgGenericToLinalgCustomBatchNorm2DNchw::checkIndexingMaps

(

ArrayAttr

array_attr

)

const

checkIteratorTypes()

bool mlir::linalg::LinalgGenericToLinalgCustomBatchNorm2DNchw::checkIteratorTypes

(

ArrayAttr

array_attr

)

const

is3dIdentityMap()

bool mlir::linalg::LinalgGenericToLinalgCustomBatchNorm2DNchw::is3dIdentityMap

(

Attribute

attr

)

const

is4dIdentityMap()

bool mlir::linalg::LinalgGenericToLinalgCustomBatchNorm2DNchw::is4dIdentityMap

(

Attribute

attr

)

const

isElementwiseRegion()

template<typename ElemOp>

bool mlir::linalg::LinalgGenericToLinalgCustomBatchNorm2DNchw::isElementwiseRegion ( Region & region ) const

inline

isTypeC()

bool mlir::linalg::LinalgGenericToLinalgCustomBatchNorm2DNchw::isTypeC	(	Type	type,
		Type	element_type,
		int64_t	c ) const

isTypeC11()

bool mlir::linalg::LinalgGenericToLinalgCustomBatchNorm2DNchw::isTypeC11	(	Type	type,
		Type	element_type,
		int64_t	c ) const

isTypeNCHW()

bool mlir::linalg::LinalgGenericToLinalgCustomBatchNorm2DNchw::isTypeNCHW	(	Type	type,
		Type	element_type,
		int64_t	n,
		int64_t	c,
		int64_t	h,
		int64_t	w ) const

matchAndRewrite()

LogicalResult mlir::linalg::LinalgGenericToLinalgCustomBatchNorm2DNchw::matchAndRewrite ( GenericOp generic_op, PatternRewriter & rewriter ) const

override

---

The documentation for this struct was generated from the following file:

• mlir/include/mlir/Dialect/Linalg/Transforms/CustomSpecializeBatchNorm.h