Searched refs:matmul_primitive (Results 1 – 1 of 1) sorted by relevance
596 std::shared_ptr<mkldnn::primitive> matmul_primitive = nullptr; in Setup() local623 matmul_primitive.reset(new mkldnn::matmul(*context_.prim_desc)); in Setup()628 context_.matmul_primitives.push_back(*matmul_primitive); in Setup()