Module args

Source

Structs§

TensorArgs
Type implementing MatmulArgs where all inputs and the output are materialized tensors.
TensorInput
Tensor input representation.
TensorInputExpand
Expand type for tensor input.
TensorInputs
Input representation for TensorArgs implementing MatmulArgs.
TensorInputsCompilationArg
TensorInputsExpand
TensorInputsLaunch
TensorMapArgs
Type implementing MatmulArgs where all inputs and the output are materialized tensor maps.
TensorMapInputs
Input representation for TensorArgs implementing MatmulArgs.
TensorMapInputsCompilationArg
TensorMapInputsExpand
TensorMapInputsLaunch
TensorOutput
Tensor output representation.
TensorOutputExpand
Expand type for tensor output.

Enums§

TensorInputIdent
Identification of the tensor input.

Traits§

ConcreteInputsFactory
Create the input runtime arguments for a matmul kernel that works on concrete inputs and output (not fused).
ConcreteOutputFactory
Create the output runtime argument for a matmul kernel that works on concrete inputs and output (not fused).
MatmulArgs
Arguments for the matrix multiplication algorithm.