| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/ |
| H A D | simple_layer_normalization.cpp | 56 const memory_desc_wrapper src_d(src_md()); in init() local
115 const memory_desc_wrapper src_d(pd()->src_md()); in execute_forward() local
135 const memory_desc_wrapper src_d(src_md()); in init() local
202 const memory_desc_wrapper src_d(pd()->src_md()); in execute_backward() local
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/ |
| H A D | simple_layer_normalization.cpp | 56 const memory_desc_wrapper src_d(src_md()); in init() local
115 const memory_desc_wrapper src_d(pd()->src_md()); in execute_forward() local
135 const memory_desc_wrapper src_d(src_md()); in init() local
203 const memory_desc_wrapper src_d(pd()->src_md()); in execute_backward() local
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ |
| H A D | gpu_inner_product_pd.hpp | 32 inline bool dense_consistency_check(const memory_desc_wrapper &src_d, in dense_consistency_check()
46 inline bool dense_gemm_consistency_check(const memory_desc_wrapper &src_d, in dense_gemm_consistency_check()
|
| /dports/math/onednn/oneDNN-2.5.1/src/gpu/ |
| H A D | gpu_inner_product_pd.hpp | 32 inline bool dense_consistency_check(const memory_desc_wrapper &src_d, in dense_consistency_check()
46 inline bool dense_gemm_consistency_check(const memory_desc_wrapper &src_d, in dense_gemm_consistency_check()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/prelu/ |
| H A D | jit_prelu_forward.cpp | 33 const memory_desc_wrapper src_d {src_md(0)}; in init() local
49 bool jit_prelu_fwd_t::pd_t::bcast_supported(const memory_desc_wrapper &src_d, in bcast_supported()
103 const memory_desc_wrapper src_d {pd()->src_md(0)}; in execute() local
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/prelu/ |
| H A D | jit_prelu_forward.cpp | 33 const memory_desc_wrapper src_d {src_md(0)}; in init() local
49 bool jit_prelu_fwd_t::pd_t::bcast_supported(const memory_desc_wrapper &src_d, in bcast_supported()
103 const memory_desc_wrapper src_d {pd()->src_md(0)}; in execute() local
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/nvidia/ |
| H A D | cudnn_gemm_inner_product.hpp | 40 inline bool gemm_consitency_check(const memory_desc_wrapper &src_d, in gemm_consitency_check()
93 inline bool reorder_check(const memory_desc_wrapper &src_d, in reorder_check()
115 inline bool dense_check(const memory_desc_wrapper &src_d, in dense_check()
|
| H A D | cudnn_matmul.cpp | 38 const auto src_d = ctx.memory_mdw(DNNL_ARG_SRC, pd()->src_md()); in execute() local
|
| /dports/math/onednn/oneDNN-2.5.1/src/gpu/nvidia/ |
| H A D | cudnn_gemm_inner_product.hpp | 40 inline bool gemm_consitency_check(const memory_desc_wrapper &src_d, in gemm_consitency_check()
93 inline bool reorder_check(const memory_desc_wrapper &src_d, in reorder_check()
115 inline bool dense_check(const memory_desc_wrapper &src_d, in dense_check()
|
| H A D | cudnn_matmul.cpp | 38 const auto src_d = ctx.memory_mdw(DNNL_ARG_SRC, pd()->src_md()); in execute() local
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ocl/ |
| H A D | gemm_matmul.cpp | 30 const auto src_d = ctx.memory_mdw(DNNL_ARG_SRC); in execute() local
|
| H A D | ref_lrn.hpp | 113 const memory_desc_wrapper src_d(pd()->src_md()); in init() local
236 const memory_desc_wrapper src_d(pd()->src_md()); in init() local
|
| /dports/math/onednn/oneDNN-2.5.1/src/gpu/ocl/ |
| H A D | gemm_matmul.cpp | 30 const auto src_d = ctx.memory_mdw(DNNL_ARG_SRC); in execute() local
|
| H A D | ref_lrn.hpp | 113 const memory_desc_wrapper src_d(pd()->src_md()); in init() local
236 const memory_desc_wrapper src_d(pd()->src_md()); in init() local
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/common/ |
| H A D | sum_pd.hpp | 122 const memory_desc_wrapper src_d(&src_mds_[i]); in init() local
144 const memory_desc_wrapper src_d(src_mds_[i]); in set_default_params() local
|
| H A D | concat_pd.hpp | 167 const memory_desc_wrapper src_d(src_mds_[i]); in set_default_params() local
202 const memory_desc_wrapper src_d(src_mds_[i]); in set_default_params() local
|
| /dports/math/onednn/oneDNN-2.5.1/src/common/ |
| H A D | sum_pd.hpp | 122 const memory_desc_wrapper src_d(&src_mds_[i]); in init() local
144 const memory_desc_wrapper src_d(src_mds_[i]); in set_default_params() local
|
| H A D | concat_pd.hpp | 167 const memory_desc_wrapper src_d(src_mds_[i]); in set_default_params() local
202 const memory_desc_wrapper src_d(src_mds_[i]); in set_default_params() local
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
| H A D | jit_avx512_core_x8s8s32x_1x1_convolution.hpp | 85 const memory_desc_t *src_d = src_md(); in init() local
175 const memory_desc_wrapper src_d(src_md); in depthwise_po_init() local
|
| H A D | jit_avx512_core_amx_deconvolution.cpp | 69 const memory_desc_wrapper src_d(pd()->src_md()); in execute_forward() local
|
| H A D | jit_uni_x8s8s32x_1x1_convolution.hpp | 87 const memory_desc_t *src_d = src_md(); in init() local
229 const memory_desc_wrapper src_d(src_md); in depthwise_po_init() local
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
| H A D | jit_avx512_core_amx_int8_deconvolution.cpp | 69 const memory_desc_wrapper src_d(pd()->src_md()); in execute_forward() local
|
| H A D | jit_avx512_core_x8s8s32x_1x1_convolution.hpp | 81 const memory_desc_t *src_d = src_md(); in init() local
204 const memory_desc_wrapper src_d(src_md); in depthwise_po_init() local
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/aarch64/ |
| H A D | jit_sve_512_x8s8s32x_convolution.cpp | 51 const memory_desc_wrapper src_d(pd()->src_md()); in execute_forward_1d() local
162 const memory_desc_wrapper src_d(pd()->src_md()); in execute_forward_2d() local
305 const memory_desc_wrapper src_d(pd()->src_md()); in execute_forward_2d_dw() local
396 const memory_desc_wrapper src_d(pd()->src_md()); in execute_forward_3d() local
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/aarch64/ |
| H A D | jit_sve_512_x8s8s32x_convolution.cpp | 51 const memory_desc_wrapper src_d(pd()->src_md()); in execute_forward_1d() local
162 const memory_desc_wrapper src_d(pd()->src_md()); in execute_forward_2d() local
305 const memory_desc_wrapper src_d(pd()->src_md()); in execute_forward_2d_dw() local
396 const memory_desc_wrapper src_d(pd()->src_md()); in execute_forward_3d() local
|