xref: /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/MatMul.h (revision 523fa7a60841cd1ecfb9cc4201f1ca8b03ed023a)
1 /*
2  * Copyright (c) Meta Platforms, Inc. and affiliates.
3  * All rights reserved.
4  *
5  * This source code is licensed under the BSD-style license found in the
6  * LICENSE file in the root directory of this source tree.
7  */
8 
9 #pragma once
10 
11 #include <executorch/backends/vulkan/runtime/graph/ComputeGraph.h>
12 
13 namespace vkcompute {
14 
15 void add_matmul_node(
16     ComputeGraph& graph,
17     const ValueRef mat1,
18     const ValueRef mat2_data,
19     const ValueRef out,
20     const ValueRef mat2_is_transposed);
21 
22 } // namespace vkcompute
23