nn.h File Reference

Attributes for neural network operators. More...

#include <tvm/relax/expr.h>

Include dependency graph for nn.h:

Go to the source code of this file.

Classes
struct	tvm::relax::Conv1DAttrs
	Attributes used in Conv1d operator. More...
struct	tvm::relax::Conv2DAttrs
	Attributes used in Conv2d operator. More...
struct	tvm::relax::Conv3DAttrs
	Attributes used in Conv3d operator. More...
struct	tvm::relax::Conv1DTransposeAttrs
	Attributes used in Conv1DTranspose operator. More...
struct	tvm::relax::Conv2DTransposeAttrs
	Attributes used in Conv2d operator. More...
struct	tvm::relax::Pool1DAttrs
	Attributes used in max_pool1d and avg_pool1d operator. More...
struct	tvm::relax::Pool2DAttrs
	Attributes used in max_pool2d and avg_pool2d operator. More...
struct	tvm::relax::Pool3DAttrs
	Attributes used in max_pool3d and avg_pool3d operator. More...
struct	tvm::relax::AdaptivePool1DAttrs
	Attributes for 1d adaptive pool operator. More...
struct	tvm::relax::AdaptivePool2DAttrs
	Attributes for 2d adaptive pool operator. More...
struct	tvm::relax::AdaptivePool3DAttrs
	Attributes for 3d adaptive pool operator. More...
struct	tvm::relax::SoftmaxAttrs
	Attributes used in softmax operators. More...
struct	tvm::relax::LeakyReluAttrs
	Attributes used in softmax operators. More...
struct	tvm::relax::BatchNormAttrs
	Attributes used in batch_norm operator. More...
struct	tvm::relax::LayerNormAttrs
	Attributes used in layer_norm operator. More...
struct	tvm::relax::GroupNormAttrs
	Attributes used in group_norm operator. More...
struct	tvm::relax::RMSNormAttrs
	Attributes used in rms_norm operator. More...
struct	tvm::relax::NLLLossAttrs
	Attributes used in nll_loss operator. More...
struct	tvm::relax::DropoutAttrs
	Attributes used in dropout operator. More...
struct	tvm::relax::AttentionAttrs
	Attributes used in Attention operator. More...
struct	tvm::relax::PadAttrs
	Attributes used for the padding operator. More...

Namespaces
	tvm
	runtime implementation for LibTorch/TorchScript.
	tvm::relax

Detailed Description

Attributes for neural network operators.