Modules§
- algorithm_
config_ proto - Nested message and enum types in
AlgorithmConfigProto. - algorithm_
proto - Nested message and enum types in
AlgorithmProto. - tensor_
descriptor_ proto - Nested message and enum types in
TensorDescriptorProto.
Structs§
- Algorithm
Config Proto - Proto definition of AlgorithmConfig in “dnn.h”. TODO(ruochengw): After cl/380702564 is submitted, add support for algorithm configs with cuDNN Frontend APIs.
- Algorithm
Proto - Generic algorithm representation.
- Convolution
Descriptor Proto - Convolution-specific parameters.
- Tensor
Descriptor Proto - Generic tensor representation.
Enums§
- Activation
Mode - Describes a kind of non-linearity (threshold-like mathematical function).
- Convolution
Kind - Convolution
Mode - Describe the math definition for the conv op. The popular behavior is actually called cross-correlation in math, despite the operation is often referred as convolution. See cuDNN cudnnConvolutionMode_t.
- Data
Layout - Describes how a convolution input or output layer’s data is formatted.
- Data
Type - Specifies the data type used by an operation.
- Filter
Layout - Describes how a convolution filter is laid out in the memory.
- Fmha
Mask Kind - FusedMHAMaskKind kind
- Fused
MhaKind - FusedMHAKind kind
- Norm
Kind - NormKind kind