Inheritance diagram for caffe2::OperatorBase:

Public Member Functions
	OperatorBase (const OperatorDef &operator_def, Workspace *ws)

bool	HasArgument (const string &name) const
	Checks if the operator has an argument of the given name.

template<typename T >
T	GetSingleArgument (const string &name, const T &default_value) const

template<typename T >
bool	HasSingleArgumentOfType (const string &name) const

template<typename T >
vector< T >	GetRepeatedArgument (const string &name, const vector< T > &default_value={}) const

template<typename T >
const T &	Input (int idx)

template<typename T >
T *	Output (int idx)

const Blob &	InputBlob (int idx)

Blob *	OutputBlob (int idx)

template<typename T >
bool	InputIsType (int idx)

template<typename T >
bool	OutputIsType (int idx)

int	InputSize ()

int	OutputSize ()

const vector< const Blob * > &	Inputs () const

const vector< Blob * > &	Outputs ()

virtual bool	Run (int stream_id=0)

virtual bool	RunAsync (int stream_id=0)

virtual void	AddRelatedBlobInfo (EnforceNotMet *err)

const OperatorDef &	def () const

const ArgumentHelper &	arg_helper () const

Detailed Description

Definition at line 23 of file operator.h.

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

Public Member Functions