Skip to main content

Mojo function

welford_block_all_reduce

welford_block_all_reduce[dtype: DType, //](thread_mean: Scalar[dtype], thread_m2: Scalar[dtype], thread_count: Scalar[dtype], mut res_mean: Scalar[dtype], mut res_m2: Scalar[dtype], mut res_count: Scalar[dtype])

Was this page helpful?