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