 |
Caffe2 - C++ API
A deep learning, cross platform ML framework
|
|
Caffe2 - C++ API
A deep learning, cross platform ML framework
|
Public Member Functions | |
| USE_RECURRENT_BASE_FUNCTIONS | RecurrentOp (const OperatorDef &operator_def, Workspace *ws) |
| bool | RunOnDevice () override |
 Public Member Functions inherited from caffe2::RecurrentBaseOp< T > | |
| USE_OPERATOR_FUNCTIONS (CUDAContext) | |
| RecurrentBaseOp (const OperatorDef &operator_def, Workspace *ws) | |
| Public Member Functions inherited from caffe2::Operator< CUDAContext > | |
| Operator (const OperatorDef &operator_def, Workspace *ws) | |
| const Tensor< CUDAContext > & | Input (int idx) |
| Tensor< CUDAContext > * | Output (int idx) |
| void | WaitEvent (const Event &ev, int stream_id=-1) final |
| void | WaitEvents (const std::vector< const Event * > &events, int stream_id=-1) final |
| bool | Run (int stream_id=0) final |
| bool | RunAsync (int stream_id=0) final |
| bool | IsStreamFree (int stream_id) const override |
| bool | HasAsyncPart () const override |
| bool | SupportsAsyncScheduling () const override |
| const CUDAContext * | getContext () const |
| 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) |
| template<typename T > | |
| T * | Output (int idx, T *allocated) |
| 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 () const |
| int | OutputSize () const |
| const vector< const Blob * > & | Inputs () const |
| const vector< Blob * > & | Outputs () |
| vector< TensorShape > | InputTensorShapes () |
| void | Wait (const OperatorBase &other, int stream_id=-1) |
| virtual void | Finish () |
| virtual void | AddRelatedBlobInfo (EnforceNotMet *err) |
| const OperatorDef & | debug_def () const |
| void | set_debug_def (const std::shared_ptr< const OperatorDef > &operator_def) |
| bool | has_debug_def () const |
| void | RecordLastFailedOpNetPosition () |
| int | net_position () const |
| void | set_net_position (int idx) |
| const DeviceOption & | device_option () const |
| const Event & | event () const |
| Event & | event () |
| void | ResetEvent () |
| void | DisableEvent () |
| bool | IsEventDisabled () const |
| const std::string & | type () const |
| void | annotate_engine (const std::string &engine) |
| const std::string & | engine () const |
| Public Member Functions inherited from caffe2::Observable< OperatorBase > | |
| const Observer * | AttachObserver (std::unique_ptr< Observer > observer) |
| std::unique_ptr< Observer > | DetachObserver (const Observer *observer_ptr) |
| Returns a unique_ptr to the removed observer. More... | |
| virtual size_t | NumObservers () |
| void | StartAllObservers () |
| void | StopAllObservers () |
Protected Member Functions | |
| INPUT_TAGS (INPUT, HIDDEN_INPUT, CELL_INPUT, WEIGHT) | |
| OUTPUT_TAGS (OUTPUT, HIDDEN_OUTPUT, CELL_OUTPUT, RNN_SCRATCH, DROPOUT_STATES) | |
| Protected Member Functions inherited from caffe2::RecurrentBaseOp< T > | |
| void | initialize (const Tensor< CUDAContext > &input, Tensor< CUDAContext > *dropoutStates=nullptr, Tensor< CUDAContext > *output=nullptr, Tensor< CUDAContext > *hiddenOutput=nullptr, Tensor< CUDAContext > *cellOutput=nullptr) |
| Protected Member Functions inherited from caffe2::Operator< CUDAContext > | |
| void | RecordEvent (const char *err_msg=nullptr) final |
| std::string | getErrorMsg () |
| Protected Member Functions inherited from caffe2::OperatorBase | |
| DISABLE_COPY_AND_ASSIGN (OperatorBase) | |
Additional Inherited Members | |
| Public Types inherited from caffe2::Observable< OperatorBase > | |
| using | Observer = ObserverBase< OperatorBase > |
| Static Public Attributes inherited from caffe2::OperatorBase | |
| static constexpr int | kNoNetPositionSet = -1 |
| Protected Attributes inherited from caffe2::RecurrentBaseOp< T > | |
| CuDNNWrapper | cudnn_wrapper_ |
| cudnnDropoutDescriptor_t | dropoutDesc_ |
| cudnnRNNDescriptor_t | rnnDesc_ |
| cudnnFilterDescriptor_t | wDesc_ |
| cudnnTensorDescriptor_t | hxDesc_ |
| cudnnTensorDescriptor_t | cxDesc_ |
| cudnnTensorDescriptor_t | hyDesc_ |
| cudnnTensorDescriptor_t | cyDesc_ |
| std::unique_ptr< detail::TensorDescriptors< T > > | xDesc_ |
| std::unique_ptr< detail::TensorDescriptors< T > > | yDesc_ |
| std::vector< TIndex > | cachedInputDims_ |
| size_t | reserveNbytes_ |
| size_t | cudnnWsNbytes_ |
| Protected Attributes inherited from caffe2::Operator< CUDAContext > | |
| CUDAContext | context_ |
| Protected Attributes inherited from caffe2::OperatorBase | |
| std::unique_ptr< Event > | event_ |
| Protected Attributes inherited from caffe2::Observable< OperatorBase > | |
| std::vector< std::unique_ptr< Observer > > | observers_list_ |
Definition at line 84 of file recurrent_op_cudnn.h.
1.8.11
