1 | #pragma once |
---|---|
2 | |
3 | #include <torch/csrc/lazy/core/ir.h> |
4 | |
5 | #include <string> |
6 | |
7 | namespace torch { |
8 | namespace lazy { |
9 | |
10 | class BackendDevice; |
11 | |
12 | class TORCH_API DumpUtil { |
13 | public: |
14 | static std::string ToDot(c10::ArrayRef<const Node*> nodes); |
15 | |
16 | static std::string PostOrderToDot( |
17 | c10::ArrayRef<const Node*> post_order, |
18 | c10::ArrayRef<const Node*> roots); |
19 | |
20 | static std::string ToText(c10::ArrayRef<const Node*> nodes); |
21 | |
22 | static std::string PostOrderToText( |
23 | c10::ArrayRef<const Node*> post_order, |
24 | c10::ArrayRef<const Node*> roots); |
25 | |
26 | static std::string ToBackend( |
27 | c10::ArrayRef<Value> values, |
28 | const BackendDevice& device); |
29 | }; |
30 | |
31 | } // namespace lazy |
32 | } // namespace torch |
33 |