pub struct OptimizedKernels;
Expand description
Collection of optimized CUDA kernels for neural operations
Implementations§
Source§impl OptimizedKernels
impl OptimizedKernels
Sourcepub fn matrix_multiply_kernel(
rows_a: usize,
cols_a: usize,
cols_b: usize,
) -> &'static str
pub fn matrix_multiply_kernel( rows_a: usize, cols_a: usize, cols_b: usize, ) -> &'static str
Get optimized matrix multiplication kernel
Sourcepub fn vector_operations_kernel() -> &'static str
pub fn vector_operations_kernel() -> &'static str
Get optimized vector operations kernel
Sourcepub fn activation_functions_kernel() -> &'static str
pub fn activation_functions_kernel() -> &'static str
Get optimized activation functions kernel
Sourcepub fn convolution_kernel() -> &'static str
pub fn convolution_kernel() -> &'static str
Get optimized convolution kernel
Sourcepub fn forward_propagation_kernel() -> &'static str
pub fn forward_propagation_kernel() -> &'static str
Get optimized forward propagation kernel
Sourcepub fn backward_propagation_kernel() -> &'static str
pub fn backward_propagation_kernel() -> &'static str
Get optimized backward propagation kernel
Sourcepub fn reduction_operations_kernel() -> &'static str
pub fn reduction_operations_kernel() -> &'static str
Get optimized reduction operations kernel
Sourcepub fn get_activation_kernel(
function: ActivationFunction,
) -> NeuralResult<String>
pub fn get_activation_kernel( function: ActivationFunction, ) -> NeuralResult<String>
Get kernel for specific activation function
Sourcepub fn get_combined_kernels() -> &'static str
pub fn get_combined_kernels() -> &'static str
Get all kernels as a single compilation unit
Auto Trait Implementations§
impl Freeze for OptimizedKernels
impl RefUnwindSafe for OptimizedKernels
impl Send for OptimizedKernels
impl Sync for OptimizedKernels
impl Unpin for OptimizedKernels
impl UnwindSafe for OptimizedKernels
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more