1 #pragma once 2 #include <functional> 3 #include <memory> 4 #include <string> 5 6 #include <torch/csrc/Export.h> 7 #include <torch/csrc/jit/ir/ir.h> 8 9 namespace torch::jit { 10 11 TORCH_API void InlineLoopCondition(std::shared_ptr<Graph>& graph); 12 TORCH_API void InlineBlockBeforeNode(Node* before_node, Block* block); 13 14 } // namespace torch::jit 15