Primitive descriptor for LBR_GRU backward 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 LBR GRU backward 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... | |
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... | |
Primitive descriptor for LBR_GRU backward propagation.
|
inline |
Initializes a primitive descriptor for LBR GRU backward 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.
|
inline |
Queries diff source iteration memory descriptor.
Returns a zero_md if no diff_src_iter was specified at op_desc creation time.
|
inline |
Queries diff destination iteration memory descriptor.
Returns a zero_md if no diff_dst_iter was specified at op_desc creation time.