tvm
|
Attributes for tensor manipulation operators. More...
Go to the source code of this file.
Classes | |
struct | tvm::relax::ConcatAttrs |
Attributes used in concat operators. More... | |
struct | tvm::relax::ExpandDimsAttrs |
Attributes used in expand_dims operators. More... | |
struct | tvm::relax::LayoutTransformAttrs |
Attributes used in layout_transform operator. More... | |
struct | tvm::relax::PermuteDimsAttrs |
Attributes used in permute_dims operator. More... | |
struct | tvm::relax::SplitAttrs |
Attributes used in split operator. More... | |
struct | tvm::relax::SqueezeAttrs |
Attributes used in squeeze operators. More... | |
struct | tvm::relax::RepeatAttrs |
Attributes used in repeat operators. More... | |
struct | tvm::relax::TileAttrs |
Attributes used in tile operators. More... | |
struct | tvm::relax::FlipAttrs |
Attributes used in flip operators. More... | |
struct | tvm::relax::GatherElementsAttrs |
Attributes used in gather_elements operators. More... | |
struct | tvm::relax::GatherNDAttrs |
Attributes used in gather_nd operators. More... | |
struct | tvm::relax::ScatterElementsAttrs |
Attributes used in scatter_elements operators. More... | |
struct | tvm::relax::ScatterNDAttrs |
Attributes used in scatter_nd operators. More... | |
struct | tvm::relax::OneHotAttrs |
Attributes used in one_hot operator. More... | |
Namespaces | |
tvm | |
runtime implementation for LibTorch/TorchScript. | |
tvm::relax | |
Attributes for tensor manipulation operators.