Primitive descriptor for RNN backward propagation. More...
#include <mkldnn.hpp>
Inheritance diagram for mkldnn::vanilla_rnn_backward::primitive_desc:
Collaboration diagram for mkldnn::vanilla_rnn_backward::primitive_desc:
Public Member Functions | |
| memory::desc | src_layer_desc () const |
| Queries source layer memory descriptor. | |
| memory::desc | src_iter_desc () const |
| Queries source iteration memory descriptor. More... | |
| memory::desc | weights_layer_desc () const |
| Queries weights layer memory descriptor. | |
| memory::desc | weights_iter_desc () const |
| Queries weights iteration memory descriptor. | |
| memory::desc | bias_desc () const |
| Queries bias memory descriptor. More... | |
| memory::desc | dst_layer_desc () const |
| Queries destination layer memory descriptor. | |
| memory::desc | dst_iter_desc () const |
| Queries destination iteration memory descriptor. More... | |
| memory::desc | workspace_desc () const |
| Queries workspace memory descriptor. More... | |
| memory::desc | diff_src_layer_desc () const |
| Queries diff source layer memory descriptor. | |
| memory::desc | diff_src_iter_desc () const |
| Queries diff source iteration memory descriptor. More... | |
| memory::desc | diff_weights_layer_desc () const |
| Queries diff weights layer memory descriptor. | |
| memory::desc | diff_weights_iter_desc () const |
| Queries diff weights iteration memory descriptor. | |
| memory::desc | diff_bias_desc () const |
| Queries diff bias memory descriptor. | |
| memory::desc | diff_dst_layer_desc () const |
| Queries diff destination layer memory descriptor. | |
| memory::desc | diff_dst_iter_desc () const |
| Queries diff destination iteration memory descriptor. More... | |
 Public Member Functions inherited from mkldnn::primitive_desc | |
| primitive_desc (const_mkldnn_op_desc_t desc, const primitive_attr *attr, const engine &e, const_mkldnn_primitive_desc_t hint_fwd_pd) | |
Creates a primitive descriptor from given op_desc, attr, engine, and optionally a hint primitive descriptor from forward propagation. More... | |
| const char * | impl_info_str () const |
| Returns implementation name. | |
| memory::dim | query_s64 (query q) const |
| Queries the memory::dim value (same as int64_t) | |
| bool | next_impl () |
| Advances the next implementation for the given op descriptor. More... | |
| memory::desc | query_md (query what, int idx=0) const |
| Queries and returns requested memory descriptor. | |
| memory::desc | scratchpad_desc () const |
| Queries scratchpad memory descriptor. More... | |
| Public Member Functions inherited from mkldnn::handle< mkldnn_primitive_desc_t > | |
| handle (mkldnn_primitive_desc_t t, bool weak=false) | |
| Constructs a C handle wrapper. More... | |
| handle () | |
| Empty constructor. More... | |
| void | reset (mkldnn_primitive_desc_t t, bool weak=false) |
| Resets the value of a C handle. More... | |
| mkldnn_primitive_desc_t | get (bool allow_emtpy=false) const |
| Returns the value of the underlying C handle. | |
Detailed Description
Primitive descriptor for RNN backward propagation.
Member Function Documentation
◆ src_iter_desc()
|
inline |
Queries source iteration memory descriptor.
Returns a zero_md if no src_iter was specified at op_desc creation time.
◆ bias_desc()
|
inline |
Queries bias memory descriptor.
Returns a zero_md if no bias was specified at op_desc creation time.
◆ dst_iter_desc()
|
inline |
Queries destination iteration memory descriptor.
Returns a zero_md if no dst_iter was specified at op_desc creation time.
◆ workspace_desc()
|
inline |
Queries workspace memory descriptor.
Returns a zero_md if no worspace is required.
◆ diff_src_iter_desc()
|
inline |
Queries diff source iteration memory descriptor.
Returns a zero_md if no diff_src_iter was specified at op_desc creation time.
◆ diff_dst_iter_desc()
|
inline |
Queries diff destination iteration memory descriptor.
Returns a zero_md if no diff_dst_iter was specified at op_desc creation time.
The documentation for this struct was generated from the following file:
- include/mkldnn.hpp