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
|
#include <torch/csrc/jit/passes/onnx/onnx_log.h>
#include <iostream>
namespace torch::jit::onnx {
namespace {
bool log_enabled = false;
std::shared_ptr<std::ostream> out;
} // namespace
bool is_log_enabled() {
return log_enabled;
}
void set_log_enabled(bool enabled) {
log_enabled = enabled;
}
void set_log_output_stream(std::shared_ptr<std::ostream> out_stream) {
out = std::move(out_stream);
}
std::ostream& _get_log_output_stream() {
return out ? *out : std::cout;
}
} // namespace torch::jit::onnx
|
