Searched refs:pr_nthr_per_grp (Results 1 – 4 of 4) sorted by relevance
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/aarch64/ |
| H A D | cpu_reducer.cpp | 516 const int pr_nthr_per_grp = balancer().nthr_per_group_ / pr_grps; in reduce_nolock() local
518 if (id_in_grp >= pr_grps * pr_nthr_per_grp) return; /* idle */ in reduce_nolock()
520 const int pr_my_grp = id_in_grp / pr_nthr_per_grp; in reduce_nolock()
521 const int pr_my_id = id_in_grp % pr_nthr_per_grp; in reduce_nolock()
534 int x_blocking = choose_x_blocking(nx, ny, pr_nthr_per_grp); in reduce_nolock()
537 balance211(ny * nx / x_blocking, pr_nthr_per_grp, pr_my_id, nxy_start, in reduce_nolock()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
| H A D | cpu_reducer.cpp | 501 const int pr_nthr_per_grp = balancer().nthr_per_group_ / pr_grps; in reduce_nolock() local
503 if (id_in_grp >= pr_grps * pr_nthr_per_grp) return; /* idle */ in reduce_nolock()
505 const int pr_my_grp = id_in_grp / pr_nthr_per_grp; in reduce_nolock()
506 const int pr_my_id = id_in_grp % pr_nthr_per_grp; in reduce_nolock()
519 int x_blocking = choose_x_blocking(nx, ny, pr_nthr_per_grp); in reduce_nolock()
522 balance211(ny * nx / x_blocking, pr_nthr_per_grp, pr_my_id, nxy_start, in reduce_nolock()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/aarch64/ |
| H A D | cpu_reducer.cpp | 516 const int pr_nthr_per_grp = balancer().nthr_per_group_ / pr_grps; in reduce_nolock() local
518 if (id_in_grp >= pr_grps * pr_nthr_per_grp) return; /* idle */ in reduce_nolock()
520 const int pr_my_grp = id_in_grp / pr_nthr_per_grp; in reduce_nolock()
521 const int pr_my_id = id_in_grp % pr_nthr_per_grp; in reduce_nolock()
534 int x_blocking = choose_x_blocking(nx, ny, pr_nthr_per_grp); in reduce_nolock()
537 balance211(ny * nx / x_blocking, pr_nthr_per_grp, pr_my_id, nxy_start, in reduce_nolock()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
| H A D | cpu_reducer.cpp | 501 const int pr_nthr_per_grp = balancer().nthr_per_group_ / pr_grps; in reduce_nolock() local
503 if (id_in_grp >= pr_grps * pr_nthr_per_grp) return; /* idle */ in reduce_nolock()
505 const int pr_my_grp = id_in_grp / pr_nthr_per_grp; in reduce_nolock()
506 const int pr_my_id = id_in_grp % pr_nthr_per_grp; in reduce_nolock()
519 int x_blocking = choose_x_blocking(nx, ny, pr_nthr_per_grp); in reduce_nolock()
522 balance211(ny * nx / x_blocking, pr_nthr_per_grp, pr_my_id, nxy_start, in reduce_nolock()
|