1 #pragma once 2 3 #include <ATen/ATen.h> 4 #include <ATen/core/ivalue.h> 5 #include <ATen/core/jit_type.h> 6 #include <ATen/core/stack.h> 7 #include <torch/csrc/Export.h> 8 #include <torch/csrc/jit/ir/ir.h> 9 10 #include <list> 11 #include <vector> 12 13 namespace torch::jit { 14 15 TORCH_API void InsertGuards(std::shared_ptr<Graph> graph); 16 17 TORCH_API void RemoveProfilingNodes(const std::shared_ptr<Graph>& graph); 18 19 } // namespace torch::jit 20