Searched refs:matmul_fwd_params (Results 1 – 1 of 1) sorted by relevance
176 void Setup(const MklDnnMatMulFwdParams& matmul_fwd_params) { in Setup() argument179 context_.src_md.reset(new memory::desc({matmul_fwd_params.src_dims}, in Setup()181 matmul_fwd_params.src_format)); in Setup()183 context_.weight_md.reset(new memory::desc({matmul_fwd_params.weight_dims}, in Setup()185 matmul_fwd_params.weight_format)); in Setup()187 context_.dst_md.reset(new memory::desc({matmul_fwd_params.dst_dims}, in Setup()189 matmul_fwd_params.dst_format)); in Setup()191 context_.bias_md.reset(new memory::desc({matmul_fwd_params.bias_dims}, in Setup()202 auto const& post_op_params = matmul_fwd_params.post_op_params; in Setup()271 context_.bias_mem.reset(new memory({{matmul_fwd_params.bias_dims}, in Setup()