1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
| #pragma once
|
| #include <torch/csrc/WindowsTorchApiMacro.h>
|
| #include <atomic>
| #include <functional>
| #include <iostream>
|
| namespace torch {
| namespace jit {
|
| using PrintHandler = void (*)(const std::string&);
|
| TORCH_API void setPrintHandler(PrintHandler ph);
| TORCH_API PrintHandler getPrintHandler();
|
| } // namespace jit
| } // namespace torch
|
|
