 |
Caffe2 - C++ API
A deep learning, cross platform ML framework
|
|
Caffe2 - C++ API
A deep learning, cross platform ML framework
|
Public Member Functions | |
| MKLOperator (const OperatorDef &operator_def, Workspace *ws) | |
| const MKLMemory< T > & | Input (int idx) |
| MKLMemory< T > * | Output (int idx) |
| bool | Run (int stream_id) final |
| virtual bool | RunOnDevice ()=0 |
| void | ExecutePrimitive () |
 Public Member Functions inherited from caffe2::OperatorBase | |
| 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 | RunAsync (int stream_id=0) |
| virtual void | AddRelatedBlobInfo (EnforceNotMet *err) |
| const OperatorDef & | def () const |
| const ArgumentHelper & | arg_helper () const |
Protected Attributes | |
| PrimitiveWrapper< T > | primitive_ |
| vector< vector< TIndex > > | input_size_cache_ |
| MKLMemory< T > | buffer_ |
| void * | resources_ [dnnResourceNumber] |
Definition at line 30 of file mkl_operator.h.
1.8.14