| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ |
| H A D | gpu_batch_normalization_pd.hpp | 44 void init_default_ws(size_t bits_per_element) override { in init_default_ws() function
53 void init_default_ws(size_t bits_per_element) override { in init_default_ws() function
|
| /dports/math/onednn/oneDNN-2.5.1/src/gpu/ |
| H A D | gpu_batch_normalization_pd.hpp | 44 void init_default_ws(size_t bits_per_element) override { in init_default_ws() function
53 void init_default_ws(size_t bits_per_element) override { in init_default_ws() function
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/ |
| H A D | ref_batch_normalization.hpp | 56 if (is_training() && fuse_norm_relu()) init_default_ws(8); in init()
96 init_default_ws(8); in init()
|
| H A D | ref_pooling.hpp | 56 init_default_ws(); in init()
100 init_default_ws(); in init()
|
| H A D | nspc_batch_normalization.hpp | 59 if (is_training() && fuse_norm_relu()) init_default_ws(8); in init()
134 init_default_ws(8); in init()
|
| H A D | ncsp_batch_normalization.hpp | 60 if (is_training() && fuse_norm_relu()) init_default_ws(8); in init()
137 init_default_ws(8); in init()
|
| H A D | nhwc_pooling.hpp | 73 init_default_ws(); in init()
147 init_default_ws(); in init()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/ |
| H A D | ref_batch_normalization.hpp | 56 if (is_training() && fuse_norm_relu()) init_default_ws(8); in init()
96 init_default_ws(8); in init()
|
| H A D | ref_pooling.hpp | 57 init_default_ws(); in init()
101 init_default_ws(); in init()
|
| H A D | nspc_batch_normalization.hpp | 59 if (is_training() && fuse_norm_relu()) init_default_ws(8); in init()
136 init_default_ws(8); in init()
|
| H A D | ncsp_batch_normalization.hpp | 60 if (is_training() && fuse_norm_relu()) init_default_ws(8); in init()
140 init_default_ws(8); in init()
|
| H A D | nhwc_pooling.hpp | 73 init_default_ws(); in init()
150 init_default_ws(); in init()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
| H A D | jit_uni_pooling.hpp | 66 init_default_ws(); in init()
130 init_default_ws(); in init()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ocl/ |
| H A D | gen9_pooling.hpp | 81 init_default_ws(s32); in init()
146 init_default_ws(data_type::s32); in init()
|
| H A D | ref_pooling.hpp | 82 init_default_ws(s32); in init()
143 init_default_ws(data_type::s32); in init()
|
| H A D | gen9_batch_normalization.hpp | 65 if (is_training() && fuse_norm_relu()) init_default_ws(8); in init()
150 init_default_ws(8); in init()
|
| H A D | ref_batch_normalization.hpp | 70 if (is_training() && fuse_norm_relu()) init_default_ws(8); in init()
157 init_default_ws(8); in init()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/aarch64/ |
| H A D | jit_uni_pooling.hpp | 66 init_default_ws(); in init()
133 init_default_ws(); in init()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/aarch64/ |
| H A D | jit_uni_pooling.hpp | 66 init_default_ws(); in init()
133 init_default_ws(); in init()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
| H A D | jit_uni_pooling.hpp | 66 init_default_ws(); in init()
131 init_default_ws(); in init()
|
| /dports/math/onednn/oneDNN-2.5.1/src/gpu/ocl/ |
| H A D | gen9_pooling.hpp | 82 init_default_ws(s32); in init()
147 init_default_ws(data_type::s32); in init()
|
| H A D | gen9_batch_normalization.hpp | 65 if (is_training() && fuse_norm_relu()) init_default_ws(8); in init()
150 init_default_ws(8); in init()
|
| H A D | ref_batch_normalization.hpp | 70 if (is_training() && fuse_norm_relu()) init_default_ws(8); in init()
157 init_default_ws(8); in init()
|
| H A D | ref_pooling.hpp | 83 init_default_ws(s32); in init()
144 init_default_ws(data_type::s32); in init()
|
| H A D | gen9_global_pooling.hpp | 63 init_default_ws(data_type::s32); in init()
|