Expand description
Configuration types for tensor operations (conv, pool, interpolate, pad, etc). Configuration types for tensor operations.
Structs§
- Attention
Module Options - Options for the attention module.
- AvgPool
Options - Average pooling options.
- Conv
Options - Convolution options.
- Conv
Transpose Options - Transposed convolution options.
- Cubic
Options - Configuration options for bicubic interpolation.
- Deform
Conv Options - Deformable convolution options.
- Grid
Sample Options - Options for grid sampling operations.
- Interpolate
Options - Interpolation options.
- MaxPool
Options - Max pooling options.
- Padded
Conv Options Deprecated - Convolution options with optional asymmetric end padding.
- Unfold
Options - Unfold operation options.
Enums§
- Causal
Alignment - Anchor of the causal diagonal for
AttentionModuleOptions::is_causal. - Coordinate
Transform Mode - Coordinate transformation mode for interpolation.
- Grid
Sample Padding Mode - Padding mode for grid sampling when coordinates are out of bounds.
- Indexing
Update Op - Computation to be used to update the existing values in indexed assignment operations (scatter/select).
- Interpolate
Mode - Algorithm used.
- Nearest
Rounding Mode - Nearest rounding mode for nearest neighbor interpolation.
- PadMode
- Padding mode for tensor pad operations.