layer_norm.h File Reference

layer normalization op constructions More...

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

Include dependency graph for layer_norm.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Namespaces
	tvm
	runtime implementation for LibTorch/TorchScript.
	tvm::topi
	tvm::topi::nn

Functions
Tensor	tvm::topi::nn::layer_norm (const Tensor &data, const Tensor &gamma, const Tensor &beta, const Array< Integer > &axis, double epsilon, std::string name="T_layer_norm", std::string tag=kInjective)
	Layer normalization. More...

Detailed Description

layer normalization op constructions