1 #pragma once 2 3 #include <torch/csrc/jit/ir/ir.h> 4 5 namespace torch::jit { 6 7 void PeepholeOptimizeONNX( 8 std::shared_ptr<Graph>& graph, 9 int opset_version, 10 bool fixed_batch_size); 11 12 } // namespace torch::jit 13