Home
last modified time | relevance | path

Searched refs:ndims_full_unroll (Results 1 – 4 of 4) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/
H A Djit_uni_reorder.cpp115 int ndims_full_unroll; member
124 int ndims_full_unroll = 0; in simple_impl_desc_init() local
131 ndims_full_unroll++; in simple_impl_desc_init()
142 if (prb.ndims - ndims_full_unroll > ndims_jit_loop_max) return false; in simple_impl_desc_init()
145 desc->ndims_full_unroll = ndims_full_unroll; in simple_impl_desc_init()
915 const int nfu = d.ndims_full_unroll; in simple_impl()
917 const int n_jit_loops = prb_.ndims - d.ndims_full_unroll; in simple_impl()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/aarch64/
H A Djit_uni_reorder.cpp78 int ndims_full_unroll; member
87 int ndims_full_unroll = 0; in simple_impl_desc_init() local
94 ndims_full_unroll++; in simple_impl_desc_init()
105 if (prb.ndims - ndims_full_unroll > ndims_jit_loop_max) return false; in simple_impl_desc_init()
108 desc->ndims_full_unroll = ndims_full_unroll; in simple_impl_desc_init()
992 const int nfu = d.ndims_full_unroll; in simple_impl()
994 const int n_jit_loops = prb_.ndims - d.ndims_full_unroll; in simple_impl()
/dports/math/onednn/oneDNN-2.5.1/src/cpu/aarch64/
H A Djit_uni_reorder.cpp78 int ndims_full_unroll; member
87 int ndims_full_unroll = 0; in simple_impl_desc_init() local
94 ndims_full_unroll++; in simple_impl_desc_init()
105 if (prb.ndims - ndims_full_unroll > ndims_jit_loop_max) return false; in simple_impl_desc_init()
108 desc->ndims_full_unroll = ndims_full_unroll; in simple_impl_desc_init()
992 const int nfu = d.ndims_full_unroll; in simple_impl()
994 const int n_jit_loops = prb_.ndims - d.ndims_full_unroll; in simple_impl()
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/
H A Djit_uni_reorder.cpp102 int ndims_full_unroll = 0; member
119 int ndims_full_unroll = 0; in simple_impl_desc_init() local
131 ndims_full_unroll = 1; in simple_impl_desc_init()
140 ndims_full_unroll++; in simple_impl_desc_init()
152 if (prb.ndims - ndims_full_unroll > ndims_jit_loop_max) return false; in simple_impl_desc_init()
155 desc->ndims_full_unroll = ndims_full_unroll; in simple_impl_desc_init()
1261 const int nfu = desc.ndims_full_unroll; in create_loops()
1343 const int n_jit_loops = prb_.ndims - d.ndims_full_unroll; in simple_impl()