mlir::linalg Namespace Reference

Classes
struct	LinalgGenericToLinalgCustomBatchNorm2DNchw
	y = (x - E[x]) / sqrt(Var[x] + eps) * gamma + beta More...
struct	LinalgGenericToLinalgSoftmax
	y[..., d, ...] = exp(x) / sum_d(exp(x_d)) More...

Functions
void	populateLinalgCustomSpecializeGenericOpsPass (RewritePatternSet &patterns)
void	populateLinalgCustomRemoveRedundantOpsPass (RewritePatternSet &patterns)
void	populateLinalgCustomFusePaddingPass (RewritePatternSet &patterns)

Function Documentation

populateLinalgCustomFusePaddingPass()

void mlir::linalg::populateLinalgCustomFusePaddingPass

(

RewritePatternSet &

patterns

)

populateLinalgCustomRemoveRedundantOpsPass()

void mlir::linalg::populateLinalgCustomRemoveRedundantOpsPass

(

RewritePatternSet &

patterns

)

populateLinalgCustomSpecializeGenericOpsPass()

void mlir::linalg::populateLinalgCustomSpecializeGenericOpsPass

(

RewritePatternSet &

patterns

)