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 vkapi::ShaderInfo get_nchw_to_tensor_shader( 16 const api::vTensor& v_dst, 17 bool int8_buffer_enabled = true); 18 vkapi::ShaderInfo get_tensor_to_nchw_shader( 19 const api::vTensor& v_src, 20 bool int8_buffer_enabled = true); 21 22 } // namespace vkcompute 23