CustomSpecializeBatchNorm.h File Reference

#include "mlir/Dialect/Linalg/IR/Linalg.h"
#include "mlir/Dialect/Tensor/IR/Tensor.h"
#include "mlir/IR/PatternMatch.h"

Go to the source code of this file.

Classes
struct	mlir::linalg::LinalgGenericToLinalgCustomBatchNorm2DNchw
	y = (x - E[x]) / sqrt(Var[x] + eps) * gamma + beta More...

Namespaces
namespace	mlir
namespace	mlir::linalg