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 14 namespace torch::jit { 15 16 TORCH_API void OptimizeFrozenGraph( 17 std::shared_ptr<Graph>& graph, 18 bool optimize_numerics = true); 19 20 } // namespace torch::jit 21