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 9 TORCH_API void transformConv1dToConv2d(std::shared_ptr<Graph>& graph); 10 TORCH_API void transformConv1dToConv2d(script::Module& module); 11 TORCH_API void insertPrePackedOps(std::shared_ptr<Graph>& graph); 12 TORCH_API void insertPrePackedOps(script::Module& module); 13 TORCH_API void fusePrePackedLinearConvWithClamp(script::Module& module); 14 TORCH_API void FoldPrePackingOps(script::Module& module); 15 TORCH_API script::Module optimizeForMobile( 16 const script::Module& module, 17 const std::set<MobileOptimizerType>& optimization_blocklist = {}, 18 const std::vector<std::string>& preserved_methods = {}); 19 } // namespace torch::jit 20