Tensor Comprehensions
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Pages
Public Member Functions | Static Public Attributes | Protected Member Functions | List of all members
caffe2::TcMatMulOp< T, Context, Engine > Class Template Reference

#include <matmul_op.h>

Inheritance diagram for caffe2::TcMatMulOp< T, Context, Engine >:
Inheritance graph
[legend]
Collaboration diagram for caffe2::TcMatMulOp< T, Context, Engine >:
Collaboration graph
[legend]

Public Member Functions

 TcMatMulOp (const caffe2::OperatorDef &operator_def, caffe2::Workspace *ws)
 
 ~TcMatMulOp () 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::TC_MATMUL
 

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::TcMatMulOp< T, Context, Engine >::TcMatMulOp ( const caffe2::OperatorDef &  operator_def,
caffe2::Workspace *  ws 
)
inline
template<typename T , class Context , class Engine = caffe2::DefaultEngine>
caffe2::TcMatMulOp< T, Context, Engine >::~TcMatMulOp ( )
inlineoverride

Member Function Documentation

template<typename T , class Context , class Engine = caffe2::DefaultEngine>
void caffe2::TcMatMulOp< 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::TcMatMulOp< T, Context, Engine >::description = tc::TC_MATMUL
static
The documentation for this class was generated from the following file: