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/api/api.h> 12 13 #include <executorch/backends/vulkan/runtime/graph/ComputeGraph.h> 14 15 #include <vector> 16 17 namespace vkcompute { 18 19 void add_transpose_view_node( 20 ComputeGraph& graph, 21 ValueRef input_ref, 22 ValueRef dim0_ref, 23 ValueRef dim1_ref, 24 ValueRef out_ref); 25 26 } // namespace vkcompute 27