|
| file | /__w/Mila/Mila/Mila/Src/Dnn/Compute/Devices/Cuda/Operations/Linear/CudaLinearOp.ixx |
| | CUDA implementation of Linear operation with two-phase cuBLASLt optimization.
|
| file | /__w/Mila/Mila/Mila/Src/Dnn/Compute/Devices/Cuda/Operations/Linear/CudaLinearOp.Dispatch.ixx |
| file | /__w/Mila/Mila/Mila/Src/Dnn/Compute/Devices/Cuda/Operations/Linear/CudaLinearOp.Plans.ixx |
| | cuBLASLt plan builders for CudaLinearOp forward and backward passes.
|
| file | /__w/Mila/Mila/Mila/Src/Dnn/Compute/Devices/Cuda/Operations/Linear/CudaLinearOp.Quantize.ixx |
| | Quantize partition of CudaLinearOp.
|