#pragma once #include namespace torch { namespace jit { TORCH_API void InlineAutodiffSubgraphs( std::shared_ptr& graph, size_t threshold = 5); } } // namespace torch