Skip to main content

gpu_div

ferrotorch_gpu::kernels

Function gpu_div 

Source 
pub fn gpu_div(
    a: &CudaBuffer<f32>,
    b: &CudaBuffer<f32>,
    device: &GpuDevice,
) -> GpuResult<CudaBuffer<f32>>
Expand description

Elementwise division: out[i] = a[i] / b[i].