jit_trace.h

#include <torch/csrc/jit/ir/ir.h>
#include <memory>

namespace torch::jit {
TORCH_API std::shared_ptr<Graph> TraceGraph(
    const std::shared_ptr<Graph>& graph,
    Stack& stack);
} // namespace torch::jit