Function burn_wgpu::kernel::binary_elemwise

source ·

pub fn binary_elemwise<K: StaticKernelSource, E: WgpuElement, const D: usize, const WORKGROUP: usize>(
    lhs: WgpuTensor<E, D>,
    rhs: WgpuTensor<E, D>
) -> WgpuTensor<E, D>

Expand description

Execute a binary kernel using the provided WORKGROUP.