Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Protected Member Functions
caffe2::RecurrentGradientOp< T > Class Template Reference
Inheritance diagram for caffe2::RecurrentGradientOp< T >:
caffe2::RecurrentBaseOp< T > caffe2::Operator< CUDAContext > caffe2::OperatorBase caffe2::Observable< OperatorBase >

Public Member Functions

USE_RECURRENT_BASE_FUNCTIONS RecurrentGradientOp (const OperatorDef &operator_def, Workspace *ws)
bool RunOnDevice () override
- Public Member Functions inherited from caffe2::RecurrentBaseOp< T >
 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 CUDAContextgetContext () 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 >
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 BlobInputBlob (int idx)
BlobOutputBlob (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 Eventevent () const
Eventevent ()
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 ObserverAttachObserver (std::unique_ptr< Observer > observer)
std::unique_ptr< ObserverDetachObserver (const Observer *observer_ptr)
 Returns a unique_ptr to the removed observer. More...
virtual size_t NumObservers ()
void StartAllObservers ()
void StopAllObservers ()

Protected Member Functions

- 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

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< Eventevent_
- Protected Attributes inherited from caffe2::Observable< OperatorBase >
std::vector< std::unique_ptr< Observer > > observers_list_

Detailed Description

template<typename T>
class caffe2::RecurrentGradientOp< T >

Definition at line 110 of file recurrent_op_cudnn.h.

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