1 #pragma once 2 3 #include <torch/csrc/jit/api/module.h> 4 #include <torch/csrc/jit/ir/ir.h> 5 #include <torch/csrc/jit/passes/mobile_optimizer_type.h> 6 7 namespace torch::jit { 8 TORCH_API void vulkanInsertPrePackedOps(std::shared_ptr<Graph>& graph); 9 TORCH_API void vulkanInsertPrePackedOps(script::Module& module); 10 TORCH_API void vulkanFusePrePackedConvWithClamp(script::Module& module); 11 TORCH_API void vulkanFoldPrePackingOps(script::Module& module); 12 TORCH_API script::Module vulkanOptimizeForMobile( 13 const script::Module& module, 14 const std::set<MobileOptimizerType>& optimization_blocklist, 15 const std::vector<std::string>& preserved_methods); 16 } // namespace torch::jit 17