1#pragma once
2
3#include <torch/csrc/jit/ir/ir.h>
4
5namespace torch {
6namespace jit {
7
8// This pass converts aten ops to a normalized form. It is
9// run immediately after IR generation in both the tracer and compiler,
10// so downstream consumers of the IR do not need handle ops in their
11// pre-normalized form.
12// Currently only handles normalization of op aliases.
13TORCH_API void NormalizeOps(const std::shared_ptr<Graph>& graph);
14
15const std::unordered_map<Symbol, Symbol>& getOperatorAliasMap();
16
17} // namespace jit
18} // namespace torch
19