xref: /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/jit_trace.h (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1 #include <torch/csrc/jit/ir/ir.h>
2 #include <memory>
3 
4 namespace torch::jit {
5 TORCH_API std::shared_ptr<Graph> TraceGraph(
6     const std::shared_ptr<Graph>& graph,
7     Stack& stack);
8 } // namespace torch::jit
9