Skip to main content

Module ops

Module ops 

Source
Expand description

Configuration types for tensor operations (conv, pool, interpolate, pad, etc). Configuration types for tensor operations.

Structs§

AttentionModuleOptions
Options for the attention module.
ConvOptions
Convolution options.
ConvTransposeOptions
Transposed convolution options.
DeformConvOptions
Deformable convolution options.
GridSampleOptions
Options for grid sampling operations.
InterpolateOptions
Interpolation options.
PaddedConvOptions
Convolution options with support for asymmetric padding.
UnfoldOptions
Unfold operation options.

Enums§

GridSamplePaddingMode
Padding mode for grid sampling when coordinates are out of bounds.
IndexingUpdateOp
Computation to be used to update the existing values in indexed assignment operations (scatter/select).
InterpolateMode
Algorithm used.
PadMode
Padding mode for tensor pad operations.