#pragma once #include #include namespace torch::jit { TORCH_API void TransformExits(std::shared_ptr& graph); } // namespace torch::jit