1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
|
#pragma once
#include <torch/csrc/Export.h>
#include <memory>
#include <ostream>
#include <string>
namespace torch {
namespace jit {
namespace onnx {
TORCH_API bool is_log_enabled();
TORCH_API void set_log_enabled(bool enabled);
TORCH_API void set_log_output_stream(std::shared_ptr<std::ostream> out_stream);
TORCH_API std::ostream& _get_log_output_stream();
#define ONNX_LOG(...) \
if (::torch::jit::onnx::is_log_enabled()) { \
::torch::jit::onnx::_get_log_output_stream() \
<< ::c10::str(__VA_ARGS__) << std::endl; \
}
} // namespace onnx
} // namespace jit
} // namespace torch
|
