Primitive descriptor for RNN forward propagation. More...
#include <dnnl.hpp>
Inherits dnnl::rnn_primitive_desc_base.
Public Member Functions | |
primitive_desc (dnnl_primitive_desc_t pd) | |
Initializes a primitive descriptor for RNN forward propagation from a C primitive descriptor pd . More... | |
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... | |
Primitive descriptor for RNN forward propagation.
|
inline |
Initializes a primitive descriptor for RNN forward propagation from a C primitive descriptor pd
.
|
inline |
Queries source iteration memory descriptor.
Returns a zero_md if no src_iter was specified at op_desc creation time.
|
inline |
Queries bias memory descriptor.
Returns a zero_md if no bias was specified at op_desc creation time.
|
inline |
Queries destination iteration memory descriptor.
Returns a zero_md if no dst_iter was specified at op_desc creation time.
|
inline |
Queries workspace memory descriptor.
Returns a zero_md if no worspace is required.