1 #pragma once 2 #include <c10/macros/Export.h> 3 #include <torch/csrc/jit/ir/ir.h> 4 #include <string> 5 #include <unordered_map> 6 7 namespace torch::jit { 8 9 TORCH_API void populate_upgraders_graph_map(); 10 11 TORCH_API std::unordered_map<std::string, std::shared_ptr<Graph>> 12 generate_upgraders_graph(); 13 14 TORCH_API std::unordered_map<std::string, std::string> get_upgraders_entry_map(); 15 16 std::shared_ptr<Graph> create_upgrader_graph( 17 const std::string& upgrader_name, 18 const std::string& upgrader_body); 19 20 } // namespace torch::jit 21