Searched refs:n_jit_loops (Results 1 – 4 of 4) sorted by relevance
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
| H A D | jit_uni_reorder.cpp | 917 const int n_jit_loops = prb_.ndims - d.ndims_full_unroll; in simple_impl() local
918 assert(n_jit_loops <= ndims_jit_loop_max); in simple_impl()
928 if (n_jit_loops > 2) loop_begin(l_loop[2], reg_cnt[2], n(nfu + 2)); in simple_impl()
930 if (n_jit_loops > 1) loop_begin(l_loop[1], reg_cnt[1], n(nfu + 1)); in simple_impl()
932 if (n_jit_loops > 0) in simple_impl()
937 if (n_jit_loops > 0) in simple_impl()
941 if (n_jit_loops > 1) in simple_impl()
945 if (n_jit_loops > 2) in simple_impl()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/aarch64/ |
| H A D | jit_uni_reorder.cpp | 994 const int n_jit_loops = prb_.ndims - d.ndims_full_unroll; in simple_impl() local
995 assert(n_jit_loops <= ndims_jit_loop_max); in simple_impl()
1009 if (n_jit_loops > 2) loop_begin(l_loop[2], reg_cnt[2], n(nfu + 2)); in simple_impl()
1011 if (n_jit_loops > 1) loop_begin(l_loop[1], reg_cnt[1], n(nfu + 1)); in simple_impl()
1013 if (n_jit_loops > 0) in simple_impl()
1022 if (n_jit_loops > 0) in simple_impl()
1026 if (n_jit_loops > 1) in simple_impl()
1030 if (n_jit_loops > 2) in simple_impl()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/aarch64/ |
| H A D | jit_uni_reorder.cpp | 994 const int n_jit_loops = prb_.ndims - d.ndims_full_unroll; in simple_impl() local
995 assert(n_jit_loops <= ndims_jit_loop_max); in simple_impl()
1009 if (n_jit_loops > 2) loop_begin(l_loop[2], reg_cnt[2], n(nfu + 2)); in simple_impl()
1011 if (n_jit_loops > 1) loop_begin(l_loop[1], reg_cnt[1], n(nfu + 1)); in simple_impl()
1013 if (n_jit_loops > 0) in simple_impl()
1022 if (n_jit_loops > 0) in simple_impl()
1026 if (n_jit_loops > 1) in simple_impl()
1030 if (n_jit_loops > 2) in simple_impl()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
| H A D | jit_uni_reorder.cpp | 1343 const int n_jit_loops = prb_.ndims - d.ndims_full_unroll; in simple_impl() local
1344 create_loops(d, reg_cnt, n_jit_loops); in simple_impl()
|