1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
| #pragma once
| #include <memory>
|
| #include <torch/csrc/WindowsTorchApiMacro.h>
|
| namespace torch {
| namespace jit {
|
| struct Graph;
|
| // Transforms loops so that they can be represented as python
| // for or while loops
| TORCH_API void CanonicalizeModifiedLoops(std::shared_ptr<Graph>& graph);
|
| } // namespace jit
| } // namespace torch
|
|