Function burn_jit::kernel::binary

source · 
pub fn binary<Kernel, KernelInplaceLhs, KernelInplaceRhs, R: Runtime, E, const D: usize>(
    lhs: JitTensor<R, E, D>,
    rhs: JitTensor<R, E, D>,
    inplace_enabled: bool,
    kernel: Kernel,
    kernel_inplace_lhs: KernelInplaceLhs,
    kernel_inplace_rhs: KernelInplaceRhs
) -> JitTensor<R, E, D>
where
    Kernel: GpuComputeShaderPhase,
    KernelInplaceLhs: GpuComputeShaderPhase,
    KernelInplaceRhs: GpuComputeShaderPhase,
    E: JitElement,
Expand description

Launch an binary operation.