For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).

Mojo function

group_norm_gpu_warp_tiling

def group_norm_gpu_warp_tiling[LayoutType: TensorLayout, origin: MutOrigin, //, dtype: DType, simd_width: Int, input_fn: def[width: Int](row: Int, col: Int) capturing thin -> SIMD[dtype, width], gamma_fn: def[width: Int](IndexList[Int(1)]) capturing thin -> SIMD[dtype, width], beta_fn: def[width: Int](IndexList[Int(1)]) capturing thin -> SIMD[dtype, width]](output: TileTensor[dtype, LayoutType, origin], epsilon: Float32, num_groups: Int32, channels_per_group: Int32, spatial: Int32)