Caffe2 - C++ API
A deep learning, cross platform ML framework
net_observer_reporter.h
1 #pragma once
2 #include <map>
3 #include "caffe2/core/net.h"
4 
5 namespace caffe2 {
6 
8  public:
9  virtual ~NetObserverReporter() = default;
10 
11  /*
12  Report the delay metric collected by the observer.
13  The delays are saved in a map. The key is an identifier associated
14  with the reported delay. The value is the delay value in float
15  */
16  virtual void reportDelay(
17  NetBase* net,
18  std::map<std::string, double>& delays,
19  const char* unit) = 0;
20 };
21 }
Copyright (c) 2016-present, Facebook, Inc.