Mojo function
welford_warp_all_reduce
welford_warp_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?
Thank you! We'll create more content like this.
Thank you for helping us improve!