Skip to main content

Module config

Module config 

Source

Structs§

DotGeneralConfig
DotGeneral dimension configuration.
GatherConfig
StableHLO gather dimension configuration.
PadConfig
StableHLO pad configuration.
ScatterConfig
StableHLO scatter dimension configuration.
SliceConfig
Slice configuration.

Enums§

CompareDir
Comparison direction.
DegenerateDotKind
The degenerate algebra a DotGeneral actually is when it is not a contraction, classified from the config and the operand shapes alone.

Functions§

degenerate_dot_kind
Classify a DotGeneral whose algebra is not a contraction.