Searched refs:desired_fmt_tag (Results 1 – 8 of 8) sorted by relevance
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/ |
| H A D | nhwc_pooling.hpp | 51 const format_tag_t desired_fmt_tag = utils::pick(ndims() - 3, in init() local
66 && memory_desc_matches_tag(*src_md(), desired_fmt_tag) in init()
67 && memory_desc_matches_tag(*dst_md(), desired_fmt_tag) in init()
127 const format_tag_t desired_fmt_tag = utils::pick(ndims() - 3, in init() local
141 && memory_desc_matches_tag(*diff_dst_md(), desired_fmt_tag) in init()
142 && memory_desc_matches_tag(*diff_src_md(), desired_fmt_tag) in init()
|
| H A D | nchw_pooling.hpp | 45 const format_tag_t desired_fmt_tag = utils::pick(ndims() - 3, in init() local
59 && memory_desc_matches_tag(*src_md(), desired_fmt_tag) in init()
60 && memory_desc_matches_tag(*dst_md(), desired_fmt_tag) in init()
107 const format_tag_t desired_fmt_tag = utils::pick(ndims() - 3, in init() local
122 && memory_desc_matches_tag(*diff_dst_md(), desired_fmt_tag) in init()
123 && memory_desc_matches_tag(*diff_src_md(), desired_fmt_tag) in init()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/ |
| H A D | nchw_pooling.hpp | 45 const format_tag_t desired_fmt_tag = utils::pick(ndims() - 3, in init() local
59 && memory_desc_matches_tag(*src_md(), desired_fmt_tag) in init()
60 && memory_desc_matches_tag(*dst_md(), desired_fmt_tag) in init()
107 const format_tag_t desired_fmt_tag = utils::pick(ndims() - 3, in init() local
122 && memory_desc_matches_tag(*diff_dst_md(), desired_fmt_tag) in init()
123 && memory_desc_matches_tag(*diff_src_md(), desired_fmt_tag) in init()
|
| H A D | nhwc_pooling.hpp | 51 const format_tag_t desired_fmt_tag = utils::pick(ndims() - 3, in init() local
66 && memory_desc_matches_tag(*src_md(), desired_fmt_tag) in init()
67 && memory_desc_matches_tag(*dst_md(), desired_fmt_tag) in init()
130 const format_tag_t desired_fmt_tag = utils::pick(ndims() - 3, in init() local
144 && memory_desc_matches_tag(*diff_dst_md(), desired_fmt_tag) in init()
145 && memory_desc_matches_tag(*diff_src_md(), desired_fmt_tag) in init()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/aarch64/ |
| H A D | jit_uni_batch_normalization_s8.cpp | 431 auto desired_fmt_tag = (ndims() == 4) ? nhwc : ndhwc; in init() local
436 && memory_desc_matches_tag(*src_md(), desired_fmt_tag) in init()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/aarch64/ |
| H A D | jit_uni_batch_normalization_s8.cpp | 431 auto desired_fmt_tag = (ndims() == 4) ? nhwc : ndhwc; in init() local
436 && memory_desc_matches_tag(*src_md(), desired_fmt_tag) in init()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
| H A D | jit_uni_batch_normalization_s8.cpp | 632 auto desired_fmt_tag = (ndims() == 4) ? nhwc : ndhwc; in init() local
637 && memory_desc_matches_tag(*src_md(), desired_fmt_tag) in init()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
| H A D | jit_uni_batch_normalization_s8.cpp | 628 auto desired_fmt_tag = (ndims() == 4) ? nhwc : ndhwc; in init() local
633 && memory_desc_matches_tag(*src_md(), desired_fmt_tag) in init()
|