Searched defs:brgemm_exec_ctx_t (Results 1 – 4 of 4) sorted by relevance
81 struct brgemm_exec_ctx_t { struct82 brgemm_exec_ctx_t(const exec_ctx_t &ctx, const pd_t *pd) in brgemm_exec_ctx_t() function91 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;
112 struct brgemm_exec_ctx_t { struct113 brgemm_exec_ctx_t(const exec_ctx_t &ctx, const pd_t *pd) in brgemm_exec_ctx_t() argument120 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;
87 struct brgemm_exec_ctx_t { struct88 brgemm_exec_ctx_t(const exec_ctx_t &ctx, const pd_t *pd) in brgemm_exec_ctx_t() function95 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;
102 struct brgemm_exec_ctx_t { struct103 brgemm_exec_ctx_t(const exec_ctx_t &ctx, const pd_t *pd) in brgemm_exec_ctx_t() function110 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;