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

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 >

Detailed Description

Definition at line 53 of file profile_observer.h.

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