Dragon - C++ API
A Computation Graph Virtual Machine Based Deep Learning Framework
Public Member Functions | Public Attributes | List of all members
dragon::OpSchema Class Reference

#include <operator_schema.h>

Public Member Functions

 OpSchema ()
 
 OpSchema (const string &op_type, const string &file, const int line)
 
bool Verify (const OperatorDef &def) const
 
OpSchemaIgnoreVerify ()
 
OpSchemaInplace (set< pair< int, int >> inplace)
 
bool AllowInplace () const
 
OpSchemaNumInputs (int n)
 
OpSchemaNumInputs (int min_num, int max_num)
 
OpSchemaNumOutputs (int n)
 
OpSchemaNumOutputs (int min_num, int max_num)
 

Public Attributes

std::function< bool(int, int)> CheckInplace
 

Constructor & Destructor Documentation

◆ OpSchema() [1/2]

dragon::OpSchema::OpSchema ( )
inline

◆ OpSchema() [2/2]

dragon::OpSchema::OpSchema ( const string &  op_type,
const string &  file,
const int  line 
)
inline

Member Function Documentation

◆ AllowInplace()

bool dragon::OpSchema::AllowInplace ( ) const
inline

◆ IgnoreVerify()

OpSchema& dragon::OpSchema::IgnoreVerify ( )
inline

◆ Inplace()

OpSchema & dragon::OpSchema::Inplace ( set< pair< int, int >>  inplace)

◆ NumInputs() [1/2]

OpSchema & dragon::OpSchema::NumInputs ( int  n)

◆ NumInputs() [2/2]

OpSchema & dragon::OpSchema::NumInputs ( int  min_num,
int  max_num 
)

◆ NumOutputs() [1/2]

OpSchema & dragon::OpSchema::NumOutputs ( int  n)

◆ NumOutputs() [2/2]

OpSchema & dragon::OpSchema::NumOutputs ( int  min_num,
int  max_num 
)

◆ Verify()

bool dragon::OpSchema::Verify ( const OperatorDef &  def) const

Member Data Documentation

◆ CheckInplace

std::function<bool(int, int)> dragon::OpSchema::CheckInplace

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