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