rms_norm.h File Reference

root mean square normalization op constructions More...

#include <tvm/te/operation.h>
#include <tvm/topi/reduction.h>
#include <tvm/topi/tags.h>
#include <string>

Include dependency graph for rms_norm.h:

Go to the source code of this file.

Namespaces
namespace	tvm
	An object that builds and maintains block scope and StmtSref mapping for Dependence analysis.
namespace	tvm::topi
namespace	tvm::topi::nn

Functions
Tensor	tvm::topi::nn::rms_norm (const Tensor &data, const Tensor &weight, const ffi::Array< int64_t > &axis, double epsilon, std::string name="T_rms_norm", std::string tag=kInjective)
	Root mean square normalization.

Detailed Description

root mean square normalization op constructions