Searched refs:mean_buf (Results 1 – 6 of 6) sorted by relevance
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/nn/ |
| H A D | layer_norm.cu | 206 AType* mean_buf = reinterpret_cast<AType*>(buf); in LayerNormFusedForwardKernelContig() local
213 mean_buf[idx] = mean; in LayerNormFusedForwardKernelContig()
220 ChanMergePartition(mean_buf[idx], sigma2_buf[idx], count_buf[idx], mean, sigma2, count); in LayerNormFusedForwardKernelContig()
226 mean_buf[threadIdx.x] = mean; in LayerNormFusedForwardKernelContig()
230 mean = mean_buf[threadIdx.x]; in LayerNormFusedForwardKernelContig()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/nn/ |
| H A D | layer_norm.cu | 206 AType* mean_buf = reinterpret_cast<AType*>(buf); in LayerNormFusedForwardKernelContig() local
213 mean_buf[idx] = mean; in LayerNormFusedForwardKernelContig()
220 ChanMergePartition(mean_buf[idx], sigma2_buf[idx], count_buf[idx], mean, sigma2, count); in LayerNormFusedForwardKernelContig()
226 mean_buf[threadIdx.x] = mean; in LayerNormFusedForwardKernelContig()
230 mean = mean_buf[threadIdx.x]; in LayerNormFusedForwardKernelContig()
|
| /dports/www/firefox-esr/firefox-91.8.0/third_party/aom/av1/encoder/ |
| H A D | rdopt.c | 4626 float mean_buf[16]; in get_mean_dev_features() local
4636 mean_buf[blk_idx] = mean; in get_mean_dev_features()
4643 get_mean_and_dev_float(mean_buf, 1, 1, blk_idx, &mean, &dev); in get_mean_dev_features()
|
| /dports/www/firefox/firefox-99.0/third_party/aom/av1/encoder/ |
| H A D | rdopt.c | 4626 float mean_buf[16]; in get_mean_dev_features() local
4636 mean_buf[blk_idx] = mean; in get_mean_dev_features()
4643 get_mean_and_dev_float(mean_buf, 1, 1, blk_idx, &mean, &dev); in get_mean_dev_features()
|
| /dports/mail/thunderbird/thunderbird-91.8.0/third_party/aom/av1/encoder/ |
| H A D | rdopt.c | 4626 float mean_buf[16]; in get_mean_dev_features() local
4636 mean_buf[blk_idx] = mean; in get_mean_dev_features()
4643 get_mean_and_dev_float(mean_buf, 1, 1, blk_idx, &mean, &dev); in get_mean_dev_features()
|
| /dports/lang/spidermonkey78/firefox-78.9.0/third_party/aom/av1/encoder/ |
| H A D | rdopt.c | 4626 float mean_buf[16]; in get_mean_dev_features() local
4636 mean_buf[blk_idx] = mean; in get_mean_dev_features()
4643 get_mean_and_dev_float(mean_buf, 1, 1, blk_idx, &mean, &dev); in get_mean_dev_features()
|