Skip to main content

Mojo function

update_w_tile_2d

update_w_tile_2d[micro_kernel_height: Int, micro_kernel_width: Int, simd_size: Int, effected_by_padding: Bool, has_residual: Bool, last_c_tile: Bool, output_dt: DType, input_dt: DType, filter_dt: DType](output: UnsafePointer[Scalar[output_dt]], input: UnsafePointer[Scalar[input_dt]], filter: UnsafePointer[Scalar[filter_dt]], _init_output: Bool, c_tile_size: Int, f_tile_offset: Int, f_tile_size: Int, conv_shape: ConvShape[2], n: Int, hw: IndexList[2])

Was this page helpful?