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
|
#include "runcnt_observer.h"
namespace caffe2 {
RunCountOperatorObserver::RunCountOperatorObserver(
OperatorBase* op,
RunCountNetObserver* netObserver)
: ObserverBase<OperatorBase>(op), netObserver_(netObserver) {
CAFFE_ENFORCE(netObserver_, "Observers can't operate outside of the net");
}
std::string RunCountNetObserver::debugInfo() {
#ifdef C10_ANDROID
// workaround
int foo = cnt_;
return "This operator runs " + c10::to_string(foo) + " times.";
#else
return "This operator runs " + c10::to_string(cnt_) + " times.";
#endif
}
void RunCountNetObserver::Start() {}
void RunCountNetObserver::Stop() {}
void RunCountOperatorObserver::Start() {
++netObserver_->cnt_;
}
void RunCountOperatorObserver::Stop() {}
std::unique_ptr<ObserverBase<OperatorBase>> RunCountOperatorObserver::rnnCopy(
OperatorBase* subject,
int rnn_order) const {
return std::unique_ptr<ObserverBase<OperatorBase>>(
new RunCountOperatorObserver(subject, netObserver_));
}
} // namespace caffe2
|
