Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Protected Attributes
caffe2::ProfileOperatorObserver Class Reference
Inheritance diagram for caffe2::ProfileOperatorObserver:
caffe2::ProfileCounter caffe2::RNNCapableOperatorObserver caffe2::ObserverBase< OperatorBase >

Public Member Functions

 ProfileOperatorObserver (OperatorBase *subject)=delete
 
 ProfileOperatorObserver (OperatorBase *subject, ProfileObserver *netObserver)
 
 ProfileOperatorObserver (OperatorBase *subject, ProfileObserver *netObserver, int net_position, int rnn_order)
 
std::unique_ptr< ObserverBase< OperatorBase > > rnnCopy (OperatorBase *subject, int rnnOrder) const override
 
void Dump () const
 
virtual std::string getId () const
 
- Public Member Functions inherited from caffe2::RNNCapableOperatorObserver
 RNNCapableOperatorObserver (OperatorBase *op)
 
- Public Member Functions inherited from caffe2::ObserverBase< OperatorBase >
 ObserverBase (OperatorBase *subject)
 
virtual std::string debugInfo ()
 
OperatorBasesubject () const
 

Protected Attributes

ProfileObservernetObserver_
 
int net_position_
 
- Protected Attributes inherited from caffe2::ProfileCounter
Timer timer_
 
float start_time_ = 0.0f
 
float run_time_ = 0.0f
 
cudaEvent_t start_
 
cudaEvent_t stop_
 
- Protected Attributes inherited from caffe2::RNNCapableOperatorObserver
int rnn_order_ = OperatorBase::kNoNetPositionSet
 
- Protected Attributes inherited from caffe2::ObserverBase< OperatorBase >
OperatorBasesubject_
 

Detailed Description

Definition at line 53 of file profile_observer.h.


The documentation for this class was generated from the following files: