rnn_capable_operator_observer.h

#pragma once

#include "caffe2/core/net.h"
#include "caffe2/core/observer.h"
#include "caffe2/core/operator.h"

namespace caffe2 {

class RNNCapableOperatorObserver : public ObserverBase<OperatorBase> {
 public:
  explicit RNNCapableOperatorObserver(OperatorBase* op)
      : ObserverBase<OperatorBase>(op){};

  virtual std::unique_ptr<ObserverBase<OperatorBase>> rnnCopy(
      OperatorBase* subject,
      int rnn_order) const = 0;
  virtual ~RNNCapableOperatorObserver(){};

 protected:
  int rnn_order_ = OperatorBase::kNoNetPositionSet;
};

} // namespace caffe2