1#pragma once
2
3#include <torch/csrc/jit/ir/ir.h>
4
5namespace torch {
6namespace jit {
7
8// Fuses Linear -> BatchNormNd into a single Linear by
9// folding batchnorm weights into linear weights.
10// This pass only works on Frozen Graphs; otherwise it is a No-Op.
11TORCH_API bool FoldFrozenLinearBatchnorm(std::shared_ptr<Graph>& graph);
12
13} // namespace jit
14} // namespace torch
15