Searched refs:ChanMergePartition (Results 1 – 2 of 2) sorted by relevance
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/nn/ |
H A D | layer_norm.cu | 75 __device__ __inline__ void ChanMergePartition(const DType lhs_mean, in ChanMergePartition() function 201 ChanMergePartition(meanB, sigma2B, countB, mean, sigma2, count); in LayerNormFusedForwardKernelContig() 220 ChanMergePartition(mean_buf[idx], sigma2_buf[idx], count_buf[idx], mean, sigma2, count); in LayerNormFusedForwardKernelContig()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/nn/ |
H A D | layer_norm.cu | 75 __device__ __inline__ void ChanMergePartition(const DType lhs_mean, in ChanMergePartition() function 201 ChanMergePartition(meanB, sigma2B, countB, mean, sigma2, count); in LayerNormFusedForwardKernelContig() 220 ChanMergePartition(mean_buf[idx], sigma2_buf[idx], count_buf[idx], mean, sigma2, count); in LayerNormFusedForwardKernelContig()
|