Home
last modified time | relevance | path

Searched defs:brgemm_exec_ctx_t (Results 1 – 4 of 4) sorted by relevance

/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/
H A Djit_brgemm_1x1_conv.hpp81 struct brgemm_exec_ctx_t { struct
82 brgemm_exec_ctx_t(const exec_ctx_t &ctx, const pd_t *pd) in brgemm_exec_ctx_t() function
91 const char *const __restrict src;
92 const char *const __restrict weights;
93 const char *const __restrict bias;
94 char *const __restrict dst;
95 const std::vector<const void *> post_ops_binary_rhs_arg_vec;
96 char *const wsp_tile;
H A Djit_brgemm_conv.hpp112 struct brgemm_exec_ctx_t { struct
113 brgemm_exec_ctx_t(const exec_ctx_t &ctx, const pd_t *pd) in brgemm_exec_ctx_t() argument
120 const char *const __restrict src;
121 const char *const __restrict weights;
122 const char *const __restrict bias;
123 char *const __restrict dst;
124 const std::vector<const void *> post_ops_binary_rhs_arg_vec;
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/
H A Djit_brgemm_1x1_conv.hpp87 struct brgemm_exec_ctx_t { struct
88 brgemm_exec_ctx_t(const exec_ctx_t &ctx, const pd_t *pd) in brgemm_exec_ctx_t() function
95 const src_data_t *const __restrict src;
96 const wei_data_t *const __restrict weights;
97 const char *const __restrict bias;
98 dst_data_t *const __restrict dst;
99 const std::vector<const void *> post_ops_binary_rhs_arg_vec;
H A Djit_brgemm_conv.hpp102 struct brgemm_exec_ctx_t { struct
103 brgemm_exec_ctx_t(const exec_ctx_t &ctx, const pd_t *pd) in brgemm_exec_ctx_t() function
110 const src_data_t *const __restrict src;
111 const wei_data_t *const __restrict weights;
112 const char *const __restrict bias;
113 dst_data_t *const __restrict dst;
114 const std::vector<const void *> post_ops_binary_rhs_arg_vec;