Tensor Comprehensions
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Pages
caffe2::TcGroupConvolutionOp< T, Context, Engine > Class Template Reference

#include <group_convolution_op.h>

Inheritance diagram for caffe2::TcGroupConvolutionOp< T, Context, Engine >:
Collaboration diagram for caffe2::TcGroupConvolutionOp< T, Context, Engine >:

Public Member Functions

 TcGroupConvolutionOp (const caffe2::OperatorDef &operator_def, caffe2::Workspace *ws)
 
 ~TcGroupConvolutionOp () override
 
- Public Member Functions inherited from caffe2::TcOp< T, Context, Engine >
 TcOp (const OperatorDef &operator_def, Workspace *ws)
 
 ~TcOp () override
 

Static Public Attributes

static constexpr auto description = tc::GROUP_CONVOLUTION2D_TC
 

Protected Member Functions

void setupNaiveMappingOptions () override
 
- Protected Member Functions inherited from caffe2::TcOp< T, Context, Engine >
virtual void setupDefaultGradMappingOptions ()
 
void prepareOutputs (const std::vector< const DLTensor * > tensorInfo)
 
virtual bool RunOnDevice () override
 

Private Attributes

int group_
 

Additional Inherited Members

- Public Attributes inherited from caffe2::TcOp< T, Context, Engine >
 USE_OPERATOR_CONTEXT_FUNCTIONS
 
- Protected Attributes inherited from caffe2::TcOp< T, Context, Engine >
std::string tc_
 
std::string gradTc_
 
std::string tcName_
 
std::string gradTcName_
 
bool profile_
 
tc::MappingOptions mappingOptions_
 
tc::MappingOptions gradMappingOptions_
 

Constructor & Destructor Documentation

template<typename T , class Context , class Engine = caffe2::DefaultEngine>
caffe2::TcGroupConvolutionOp< T, Context, Engine >::TcGroupConvolutionOp ( const caffe2::OperatorDef &  operator_def,
caffe2::Workspace *  ws 
)
inline
template<typename T , class Context , class Engine = caffe2::DefaultEngine>
caffe2::TcGroupConvolutionOp< T, Context, Engine >::~TcGroupConvolutionOp ( )
inlineoverride

Member Function Documentation

template<typename T , class Context , class Engine = caffe2::DefaultEngine>
void caffe2::TcGroupConvolutionOp< T, Context, Engine >::setupNaiveMappingOptions ( )
inlineoverrideprotectedvirtual

Hook called when the mappingOptions are not provided in the Caffe2 operator arguments. Does nothing by default, derived classes can reimplement this to customize stategies.

Reimplemented from caffe2::TcOp< T, Context, Engine >.

Member Data Documentation

template<typename T , class Context , class Engine = caffe2::DefaultEngine>
constexpr auto caffe2::TcGroupConvolutionOp< T, Context, Engine >::description = tc::GROUP_CONVOLUTION2D_TC
static
template<typename T , class Context , class Engine = caffe2::DefaultEngine>
int caffe2::TcGroupConvolutionOp< T, Context, Engine >::group_
private

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