Name	Size	Mode	Actions
operator_attaching_net_observer.h	907	0644	editdlrm
profile_observer.h	3260	0644	editdlrm
runcnt_observer.h	1319	0644	editdlrm
time_observer.h	1827	0644	editdlrm

#pragma once #include "caffe2/core/net.h" #include "caffe2/core/observer.h" #include "caffe2/core/operator.h" #include "caffe2/observers/operator_attaching_net_observer.h" namespace caffe2 { class RunCountNetObserver; class TORCH_API RunCountOperatorObserver final : public ObserverBase { public: explicit RunCountOperatorObserver(OperatorBase* op) = delete; RunCountOperatorObserver(OperatorBase* op, RunCountNetObserver* netObserver); ~RunCountOperatorObserver() {} std::unique_ptr> rnnCopy( OperatorBase* subject, int rnn_order) const override; private: void Start() override; void Stop() override; private: RunCountNetObserver* netObserver_; }; class TORCH_API RunCountNetObserver final : public OperatorAttachingNetObserver< RunCountOperatorObserver, RunCountNetObserver> { public: explicit RunCountNetObserver(NetBase* subject_) : OperatorAttachingNetObserver< RunCountOperatorObserver, RunCountNetObserver>(subject_, this), cnt_(0) {} ~RunCountNetObserver() {} std::string debugInfo() override; friend class RunCountOperatorObserver; private: void Start() override; void Stop() override; protected: std::atomic cnt_; }; } // namespace caffe2