1 #pragma once 2 3 #include <torch/csrc/jit/ir/ir.h> 4 5 namespace torch::jit { 6 7 // Peephole Optimizes alias sensitive peepholes 8 // Currently this is invoked as part of PeepholeOptimize 9 // return true if graph is modified 10 // Optimizes on TensorType if shape_peepholes is true 11 TORCH_API bool PeepholeOptimizeAliasSensitive( 12 const std::shared_ptr<Graph>& graph, 13 bool shape_peepholes); 14 15 } // namespace torch::jit 16