nn.h File Reference

Auxiliary attributes for nn operators. More...

#include <tvm/ir/attrs.h>
#include <tvm/relay/base.h>
#include <string>

Include dependency graph for nn.h:

Go to the source code of this file.

Classes
struct	tvm::relay::BiasAddAttrs
	Add a 1D Tensor to an axis of a data. More...
struct	tvm::relay::Conv1DAttrs
	Attributes used in 1D convolution operators. More...
struct	tvm::relay::Conv2DAttrs
	Attributes used in convolution operators. More...
struct	tvm::relay::ConvWinogradWeightTransformAttrs
	Attributes used in winograd weight transformation operators. More...
struct	tvm::relay::ConvGemmWeightTransformAttrs
	Attributes used in gemm weight transformation operators. More...
struct	tvm::relay::Conv2DWinogradAttrs
	Attributes used in convolution operators with winograd algorithm. More...
struct	tvm::relay::Conv2DWinogradNNPACKWeightTransformAttrs
	Attributes used in winograd weight transformation operators. More...
struct	tvm::relay::Conv3DAttrs
	Attributes used in convolution operators. More...
struct	tvm::relay::Conv3DTransposeAttrs
	Attributes used in transposed convolution operator. More...
struct	tvm::relay::Conv3DWinogradAttrs
	Attributes used in 3d winograd convolution operators. More...
struct	tvm::relay::SoftmaxAttrs
	Attributes used in softmax operators. More...
struct	tvm::relay::Conv2DTransposeAttrs
	Attributes used in transposed convolution operator. More...
struct	tvm::relay::DilateAttrs
	Attributes used in dilate operator. More...
struct	tvm::relay::Conv1DTransposeAttrs
	Attributes used in 1D transposed convolution operator. More...
struct	tvm::relay::MaxPool2DAttrs
	Attributes for max pool operator. More...
struct	tvm::relay::AvgPool2DAttrs
	Attributes for avg pool operator. More...
struct	tvm::relay::GlobalPool2DAttrs
	Attributes for global pool operator. More...
struct	tvm::relay::AdaptivePool1DAttrs
	Attributes for 1d adaptive pool operator. More...
struct	tvm::relay::AdaptivePool2DAttrs
	Attributes for 2d adaptive pool operator. More...
struct	tvm::relay::AdaptivePool3DAttrs
	Attributes for 3d adaptive pool operator. More...
struct	tvm::relay::MaxPool1DAttrs
	Attributes for 1D max pool operator. More...
struct	tvm::relay::AvgPool1DAttrs
	Attributes for 1D avg pool operator. More...
struct	tvm::relay::MaxPool3DAttrs
	Attributes for 3D max pool operator. More...
struct	tvm::relay::AvgPool3DAttrs
	Attributes for 3D avg pool operator. More...
struct	tvm::relay::MatmulAttrs
	Attributes for matmul operator. More...
struct	tvm::relay::DenseAttrs
	Attributes for dense operator. More...
struct	tvm::relay::DensePackAttrs
	Attributes for dense_pack operator. More...
struct	tvm::relay::BatchMatmulAttrs
	Attributes for batch matmul operator. More...
struct	tvm::relay::SparseDenseAttrs
	Attributes for sparse_dense operator. More...
struct	tvm::relay::SparseTransposeAttrs
	Attributes for sparse_transpose operator. More...
struct	tvm::relay::SparseConv2DAttrs
	Attributes for sparse_dense operator. More...
struct	tvm::relay::FIFOBufferAttrs
	Attributes for FIFO buffer operator. More...
struct	tvm::relay::UpSamplingAttrs
	Attributes for upsampling operator. More...
struct	tvm::relay::UpSampling3DAttrs
	Attributes for upsampling3d operator. More...
struct	tvm::relay::PadAttrs
	Attributes used for the padding operator. More...
struct	tvm::relay::MirrorPadAttrs
	Attributes used for the MirrorPadding operator. More...
struct	tvm::relay::LeakyReluAttrs
	Attributes for leaky relu operator. More...
struct	tvm::relay::PReluAttrs
	Attributes for prelu operator. More...
struct	tvm::relay::DropoutAttrs
	Attributes used in dropout operator. More...
struct	tvm::relay::BatchNormAttrs
	Attributes used in batch_norm operator. More...
struct	tvm::relay::InstanceNormAttrs
	Attributes used in instance_norm operator. More...
struct	tvm::relay::LayerNormAttrs
	Attributes used in layer_norm operator. More...
struct	tvm::relay::GroupNormAttrs
	Attributes used in group_norm operator. More...
struct	tvm::relay::LRNAttrs
	Attributes for LRN operator. More...
struct	tvm::relay::L2NormalizeAttrs
	Attributes for L2Normalize operator. More...
struct	tvm::relay::DeformableConv2DAttrs
	Attributes for DeformableConv2D operator. More...
struct	tvm::relay::SubPixelAttrs
	Attributes used in subpixel operators. More...
struct	tvm::relay::CorrelationAttrs
	Attributes used in correlation operators. More...
struct	tvm::relay::SpaceToBatchNDAttrs
	Attributes used in SpaceToBatchND operator. More...
struct	tvm::relay::BatchToSpaceNDAttrs
	Attributes used in BatchToSpaceND operator. More...
struct	tvm::relay::NLLLossAttrs
	Attributes used in NLLLoss operator. More...

Namespaces
	tvm
	runtime implementation for LibTorch/TorchScript.
	tvm::relay
	Relay: a high level functional IR for TVM.

Detailed Description

Auxiliary attributes for nn operators.