1 #pragma once 2 3 #include <memory> 4 5 #include <torch/csrc/jit/ir/ir.h> 6 7 namespace torch::jit { 8 9 void EvalPeepholeONNX( 10 std::shared_ptr<Graph>& g, 11 std::map<std::string, IValue>& paramDict); 12 13 } // namespace torch::jit 14