mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 05:34:18 +08:00
Follows #133399 Pull Request resolved: https://github.com/pytorch/pytorch/pull/134537 Approved by: https://github.com/Skylion007
9 lines
207 B
C++
9 lines
207 B
C++
#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
|