CustomSpecializeSoftmax.h File Reference

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

Go to the source code of this file.

Classes
struct	mlir::linalg::LinalgGenericToLinalgSoftmax
	y[..., d, ...] = exp(x) / sum_d(exp(x_d)) More...

Namespaces
namespace	mlir
namespace	mlir::linalg