1#pragma once
2
3#include <torch/csrc/jit/ir/ir.h>
4
5/** \brief Runs a set of Optimizations that Optimize Frozen Graphs
6 *
7 * Currently this set of optimizations is:
8 * - FoldFrozenConvBatchnorm
9 * - FoldFrozenConvAddOrSub
10 * - FoldFrozenConvMulOrDiv
11 * - FoldFrozenLinearBatchnorm
12 */
13
14namespace torch {
15namespace jit {
16
17TORCH_API void OptimizeFrozenGraph(
18 std::shared_ptr<Graph>& graph,
19 bool optimize_numerics = true);
20
21} // namespace jit
22} // namespace torch
23