Searched refs:matmul_fwd (Results 1 – 2 of 2) sorted by relevance
89 if (context_.matmul_fwd == nullptr) { in MklDnnMatMulFwdPrimitive()157 std::shared_ptr<mkldnn::primitive> matmul_fwd; member173 matmul_fwd(nullptr) {} in MklDnnMatMulFwdContext()277 context_.matmul_fwd.reset(new inner_product_forward(*context_.fwd_pd)); in Setup()283 context_.fwd_primitives.push_back(*context_.matmul_fwd); in Setup()296 MklDnnMatMulFwdPrimitive<T, Tinput, Tweight, Tbias, Toutput>* matmul_fwd = in Get() local301 matmul_fwd = in Get()306 matmul_fwd = dynamic_cast< in Get()311 if (matmul_fwd == nullptr) { in Get()312 matmul_fwd = in Get()[all …]
224 matmul_fwd = nullptr; in Compute() local234 matmul_fwd = in Compute()240 matmul_fwd_pd = matmul_fwd->GetPrimitiveDesc(); in Compute()292 cpu_stream.reset(CreateStream(context, matmul_fwd->GetEngine())); in Compute()296 matmul_fwd->Execute(src_data, weight_data, bias_data, dst_data, in Compute()