Dragon - C++ API
A Computation Graph Virtual Machine Based Deep Learning Framework
Public Member Functions | Public Attributes | Protected Attributes | List of all members
dragon::ProposalOp< Context > Class Template Referencefinal

#include <proposal_op.h>

Inheritance diagram for dragon::ProposalOp< Context >:
dragon::Operator< Context > dragon::OperatorBase

Public Member Functions

 ProposalOp (const OperatorDef &def, Workspace *ws)
 
void RunOnDevice () override
 Implement the detailed execution. More...
 
template<typename T >
void RCNNImpl ()
 
template<typename T >
void RetinaNetImpl ()
 
- Public Member Functions inherited from dragon::Operator< Context >
 Operator (const OperatorDef &def, Workspace *ws)
 Default constructor. More...
 
void Run (int stream_id=0) final
 Run this operator on the specified stream. More...
 
virtual void PrepareResource ()
 Prepare the content of inputs. More...
 
virtual void ReleaseResource ()
 Release the ownership of inputs. More...
 
virtual void MemorySwitch ()
 Coordinate the context of inputs and outputs. More...
 
Context * ctx ()
 Return the internal context. More...
 
bool AllowRun ()
 Whether this operator can be ignored. More...
 
- Public Member Functions inherited from dragon::OperatorBase
 OperatorBase (const OperatorDef &def, Workspace *ws)
 Default constructor. More...
 
virtual ~OperatorBase ()
 Default deconstructor. More...
 
TensorX (int i)
 Return the specified input tensor. More...
 
TensorY (int i)
 Return the specified output tensor. More...
 
int XSize ()
 Return the number of inputs. More...
 
int YSize ()
 Return the number of outputs. More...
 
void UpdateFrom (const OperatorDef &def)
 Modify operator according to the given def. More...
 
void SwitchToPhase (const string &phase)
 Switch the internal running phase. More...
 
virtual void Fusion (void *graph)
 Fusion this operator into the specified graph. More...
 
const string & name () const
 Return the operator name. More...
 
const string & type () const
 Return the operator type. More...
 
const string & phase () const
 Return the current running phase. More...
 
const string & handle () const
 Return the resource handle. More...
 
const string & dtype () const
 Return the data type. More...
 
const string & data_format () const
 Return the data format. More...
 
const string unique_name (const string &name) const
 Return the unique name in this operator. More...
 
Workspacews () const
 Return the parent workspace. More...
 
template<typename T >
Arg (const string &name, const T &default_value)
 Return the value of the specified argument. More...
 
template<typename T >
vector< T > Args (const string &name)
 Return the values of the specified argument. More...
 
const Map< std::string, const Argument * > & args ()
 Return the argument map. More...
 
const Argument & arg (const string &name)
 Return the specified argument. More...
 
SubGraphsubgraph ()
 Return the recomputing subgraph. More...
 
void set_subgraph (SubGraph subgraph)
 Set the given recomputing subgraph. More...
 
const OperatorDef & def () const
 Return the stored def. More...
 
string DebugString () const
 Return the debug string of stored def. More...
 
string DTypeString (const Tensor &, const Set< string > &) const
 Return the dtype string according to given tensor. More...
 
string DTypeString (const string &, const Set< string > &) const
 

Public Attributes

 USE_OPERATOR_FUNCTIONS
 

Protected Attributes

string det_type_
 
float nms_thr_
 
float score_thr_
 
vec64_t strides_
 
vec64_t indices_
 
vec64_t roi_indices_
 
vector< float > ratios_
 
vector< float > scales_
 
vector< float > scores_
 
vector< float > anchors_
 
int64_t min_size_
 
int64_t pre_nms_topn_
 
int64_t post_nms_topn_
 
int64_t num_images_
 
int64_t min_level_
 
int64_t max_level_
 
int64_t canonical_level_
 
int64_t canonical_scale_
 
Tensor proposals_
 
Tensor nms_mask_
 
- Protected Attributes inherited from dragon::Operator< Context >
Context ctx_
 Store the internal context. More...
 
bool allow_run_
 
bool allow_recomp_
 
bool do_sync_
 
- Protected Attributes inherited from dragon::OperatorBase
Workspacews_
 Store the parent workspace. More...
 
OperatorDef def_
 Store the def. More...
 
SubGraph subgraph_
 Store the recomputing subgraph. More...
 
string phase_
 Store the phase and handle. More...
 
string handle_
 
string dtype_
 Store the data type and format. More...
 
string data_format_
 
vector< Tensor * > inputs_
 Store the pointer of inputs and outputs. More...
 
vector< Tensor * > outputs_
 
Map< string, const Argument * > args_
 Store the defined arguments. More...
 

Additional Inherited Members

- Public Types inherited from dragon::OperatorBase
typedef Map< string, vector< OperatorBase * > > SubGraph
 

Constructor & Destructor Documentation

◆ ProposalOp()

template<class Context >
dragon::ProposalOp< Context >::ProposalOp ( const OperatorDef &  def,
Workspace ws 
)
inline

Member Function Documentation

◆ RCNNImpl()

template<class Context >
template<typename T >
void dragon::ProposalOp< Context >::RCNNImpl ( )

◆ RetinaNetImpl()

template<class Context >
template<typename T >
void dragon::ProposalOp< Context >::RetinaNetImpl ( )

◆ RunOnDevice()

template<class Context >
void dragon::ProposalOp< Context >::RunOnDevice ( )
overridevirtual

Implement the detailed execution.

Implements dragon::Operator< Context >.

Member Data Documentation

◆ anchors_

template<class Context >
vector<float> dragon::ProposalOp< Context >::anchors_
protected

◆ canonical_level_

template<class Context >
int64_t dragon::ProposalOp< Context >::canonical_level_
protected

◆ canonical_scale_

template<class Context >
int64_t dragon::ProposalOp< Context >::canonical_scale_
protected

◆ det_type_

template<class Context >
string dragon::ProposalOp< Context >::det_type_
protected

◆ indices_

template<class Context >
vec64_t dragon::ProposalOp< Context >::indices_
protected

◆ max_level_

template<class Context >
int64_t dragon::ProposalOp< Context >::max_level_
protected

◆ min_level_

template<class Context >
int64_t dragon::ProposalOp< Context >::min_level_
protected

◆ min_size_

template<class Context >
int64_t dragon::ProposalOp< Context >::min_size_
protected

◆ nms_mask_

template<class Context >
Tensor dragon::ProposalOp< Context >::nms_mask_
protected

◆ nms_thr_

template<class Context >
float dragon::ProposalOp< Context >::nms_thr_
protected

◆ num_images_

template<class Context >
int64_t dragon::ProposalOp< Context >::num_images_
protected

◆ post_nms_topn_

template<class Context >
int64_t dragon::ProposalOp< Context >::post_nms_topn_
protected

◆ pre_nms_topn_

template<class Context >
int64_t dragon::ProposalOp< Context >::pre_nms_topn_
protected

◆ proposals_

template<class Context >
Tensor dragon::ProposalOp< Context >::proposals_
protected

◆ ratios_

template<class Context >
vector<float> dragon::ProposalOp< Context >::ratios_
protected

◆ roi_indices_

template<class Context >
vec64_t dragon::ProposalOp< Context >::roi_indices_
protected

◆ scales_

template<class Context >
vector<float> dragon::ProposalOp< Context >::scales_
protected

◆ score_thr_

template<class Context >
float dragon::ProposalOp< Context >::score_thr_
protected

◆ scores_

template<class Context >
vector<float> dragon::ProposalOp< Context >::scores_
protected

◆ strides_

template<class Context >
vec64_t dragon::ProposalOp< Context >::strides_
protected

◆ USE_OPERATOR_FUNCTIONS

template<class Context >
dragon::ProposalOp< Context >::USE_OPERATOR_FUNCTIONS

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