Skip to main content

Module matmul

burn_std::tensor

Module matmul 

Source
Expand description

Batched matmul transformation utilities.

Structs§

MatmulTransformAnalysis
The facts of a batched matmul lhs @ rhs relevant to folding its batches into its rows.

Enums§

MatmulTransformAction
The action to take for a batched matmul with a broadcast rhs.
MatmulTransformPolicy
Decides the action to take for a batched matmul, given its analysis.