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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
|
#include "smart_tensor_printer.h"
#include "caffe2/core/operator.h"
namespace caffe2 {
namespace {
// Since DispatchHelper doesn't support passing arguments through the call()
// method to DoRunWithType we have to create an object that will hold these
// arguments explicitly.
struct ProxyPrinter {
template <typename T>
bool DoRunWithType() {
tensorPrinter->Print<T>(*tensor);
return true;
}
void Print() {
// Pulled in printable types from caffe2/core/types.cc
// Unfortunately right now one has to add them by hand
DispatchHelper<TensorTypes<
float,
int,
std::string,
bool,
uint8_t,
int8_t,
uint16_t,
int16_t,
int64_t,
double,
char>>::call(this, tensor->dtype());
}
const Tensor* tensor;
TensorPrinter* tensorPrinter;
};
} // namespace
SmartTensorPrinter::SmartTensorPrinter(const std::string& tensor_name)
: tensorPrinter_(tensor_name) {}
SmartTensorPrinter::SmartTensorPrinter(
const std::string& tensor_name,
const std::string& file_name)
: tensorPrinter_(tensor_name, file_name) {}
SmartTensorPrinter::SmartTensorPrinter(
const std::string& tensor_name,
const std::string& file_name,
int limit)
: tensorPrinter_(tensor_name, file_name, limit) {}
void SmartTensorPrinter::Print(const Tensor& tensor) {
// NOLINTNEXTLINE(cppcoreguidelines-pro-type-member-init)
ProxyPrinter printer;
printer.tensor = &tensor;
printer.tensorPrinter = &tensorPrinter_;
printer.Print();
}
SmartTensorPrinter& SmartTensorPrinter::DefaultTensorPrinter() {
// TODO(janusz): thread_local does not work under mac.
#if defined(__APPLE__)
CAFFE_THROW(
"SmartTensorPrinter does not work on mac yet due to thread_local.");
#else
static thread_local SmartTensorPrinter printer;
return printer;
#endif
}
void SmartTensorPrinter::PrintTensor(const Tensor& tensor) {
DefaultTensorPrinter().Print(tensor);
}
} // namespace caffe2
|
