elementwise_wgsl

numrs::backend::webgpu::codegen

Function elementwise_wgsl 

Source 
pub fn elementwise_wgsl(
    kind: ElementwiseKind,
    _inputs: Vec<usize>,
    output_shape: Vec<usize>,
) -> String
Expand description

Generate a simple WGSL elementwise kernel for Add/Mul