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_softmax_node( 20 ComputeGraph& graph, 21 ValueRef in, 22 ValueRef dim, 23 ValueRef out, 24 bool log_softmax); 25 26 } // namespace vkcompute 27