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 namespace vkcompute { 16 17 void add_repeat_interleave_node( 18 ComputeGraph& graph, 19 const ValueRef in, 20 const ValueRef num_repeats, 21 const ValueRef dim, 22 const ValueRef out); 23 24 } // namespace vkcompute 25