caffe2::TcConvolutionOp< T, Context, Engine > Class Template Reference

#include <convolution_op.h>

Inheritance diagram for caffe2::TcConvolutionOp< T, Context, Engine >:

Collaboration diagram for caffe2::TcConvolutionOp< T, Context, Engine >:

Public Member Functions
	TcConvolutionOp (const caffe2::OperatorDef &operator_def, caffe2::Workspace *ws)
	~TcConvolutionOp () 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::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

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::TcConvolutionOp< T, Context, Engine >::TcConvolutionOp ( const caffe2::OperatorDef &  operator_def, caffe2::Workspace *  ws  )

inline

template<typename T , class Context , class Engine = caffe2::DefaultEngine>

caffe2::TcConvolutionOp< T, Context, Engine >::~TcConvolutionOp ( )

inlineoverride

Member Function Documentation

template<typename T , class Context , class Engine = caffe2::DefaultEngine>

void caffe2::TcConvolutionOp< 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::TcConvolutionOp< T, Context, Engine >::description = tc::CONVOLUTION2D_TC

static

---

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

• include/tc/c2/convolution_op.h