1#pragma once
2
3#include <torch/csrc/jit/ir/ir.h>
4
5namespace torch {
6namespace jit {
7
8// return true if graph is modified
9TORCH_API bool PeepholeOptimize(
10 const std::shared_ptr<Graph>& graph,
11 bool disable_shape_peepholes = false);
12// return true if graph is modified
13TORCH_API bool PeepholeOptimize(
14 Block* block,
15 bool disable_shape_peepholes = false);
16// return true if graph is modified
17TORCH_API bool FuseAddMM(const std::shared_ptr<Graph>& graph);
18
19} // namespace jit
20} // namespace torch
21