1 #pragma once 2 3 #include <torch/csrc/jit/ir/ir.h> 4 5 namespace torch::jit { 6 7 // Find the valid upgrader graph for the upgrader and cache the result 8 // for later lookups. Will error out if there is no valid upgrader graph 9 // provided for the upgrader name. 10 std::shared_ptr<Graph> getUpgraderGraph(const std::string& upgrader_name); 11 12 TORCH_API void ReplaceOldOperatorsWithUpgraders(std::shared_ptr<Graph> graph); 13 14 } // namespace torch::jit 15