Matrix multiplication. GPU backends override to use accelerated matmul. Default: delegates to candle's CPU/CUDA/Metal matmul.
(&self, a: &Tensor, b: &Tensor)
| 183 | /// Matrix multiplication. GPU backends override to use accelerated matmul. |
| 184 | /// Default: delegates to candle's CPU/CUDA/Metal matmul. |
| 185 | fn matmul(&self, a: &Tensor, b: &Tensor) -> Result<Tensor> { |
| 186 | a.matmul(b) |
| 187 | } |
| 188 | |
| 189 | /// Pre-process a linear weight for optimal matmul performance. |
| 190 | /// Called once at model load time. Backends can pre-convert dtype, |
no outgoing calls
no test coverage detected