Searched refs:ou_blocks (Results 1 – 7 of 7) sorted by relevance
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/common/ |
| H A D | dnnl_debug.cpp | 92 dims_t ou_blocks = {0}; in dnnl_md2fmt_str() local
93 utils::array_copy(ou_blocks, md.padded_dims(), md.ndims()); in dnnl_md2fmt_str()
99 ou_blocks[d] /= blocks[d]; in dnnl_md2fmt_str()
105 utils::simultaneous_sort(strides, ou_blocks, dim_chars, md.ndims(), in dnnl_md2fmt_str()
|
| H A D | type_helpers.hpp | 730 dims_t ou_blocks = {0}; in memory_desc_init_by_blocking_desc() local
731 utils::array_copy(ou_blocks, md.padded_dims, ndims); in memory_desc_init_by_blocking_desc()
736 ou_blocks[d] /= blocks[d]; in memory_desc_init_by_blocking_desc()
739 utils::simultaneous_sort(mblk.strides, ou_blocks, perm, ndims, in memory_desc_init_by_blocking_desc()
|
| H A D | verbose.cpp | 246 dims_t ou_blocks = {0}; in md2fmt_str() local
247 utils::array_copy(ou_blocks, mdw.padded_dims(), mdw.ndims()); in md2fmt_str()
253 ou_blocks[d] /= blocks[d]; in md2fmt_str()
259 utils::simultaneous_sort(strides, ou_blocks, dim_chars, mdw.ndims(), in md2fmt_str()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/ |
| H A D | simple_concat.hpp | 123 dims_t ou_blocks = {0}; in format_perm() local
124 utils::array_copy(ou_blocks, dst_d.padded_dims(), ndims); in format_perm()
128 ou_blocks[d] /= blocks[d]; in format_perm()
131 utils::simultaneous_sort(strides, ou_blocks, iperm_, ndims, in format_perm()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/ |
| H A D | simple_concat.hpp | 123 dims_t ou_blocks = {0}; in format_perm() local
124 utils::array_copy(ou_blocks, dst_d.padded_dims(), ndims); in format_perm()
128 ou_blocks[d] /= blocks[d]; in format_perm()
131 utils::simultaneous_sort(strides, ou_blocks, iperm_, ndims, in format_perm()
|
| /dports/math/onednn/oneDNN-2.5.1/src/common/ |
| H A D | type_helpers.hpp | 709 dims_t ou_blocks = {0}; in memory_desc_init_by_blocking_desc() local
710 utils::array_copy(ou_blocks, md.padded_dims, ndims); in memory_desc_init_by_blocking_desc()
715 ou_blocks[d] /= blocks[d]; in memory_desc_init_by_blocking_desc()
719 mblk.strides, ou_blocks, perm, ndims, [](stride_t a, stride_t b) { in memory_desc_init_by_blocking_desc()
|
| H A D | verbose.cpp | 225 dims_t ou_blocks = {0}; in md2fmt_tag_str() local
226 utils::array_copy(ou_blocks, mdw.padded_dims(), mdw.ndims()); in md2fmt_tag_str()
232 ou_blocks[d] /= blocks[d]; in md2fmt_tag_str()
241 utils::simultaneous_sort(strides, ou_blocks, dim_chars, mdw.ndims(), in md2fmt_tag_str()
|