Skip to main content

init_forward_kernel_cache

entrenar::autograd::cuda_forward

Function init_forward_kernel_cache 

Source 
pub fn init_forward_kernel_cache(ctx: Arc<CudaContext>) -> Result<()>
Expand description

Initialize forward kernel cache with CUDA context