GRU descriptor for backward propagation. More...
#include <dnnl.hpp>
Collaboration diagram for dnnl::gru_backward::desc:
Public Member Functions | |
| desc (prop_kind aprop_kind, rnn_direction direction, const memory::desc &src_layer_desc, const memory::desc &src_iter_desc, const memory::desc &weights_layer_desc, const memory::desc &weights_iter_desc, const memory::desc &bias_desc, const memory::desc &dst_layer_desc, const memory::desc &dst_iter_desc, const memory::desc &diff_src_layer_desc, const memory::desc &diff_src_iter_desc, const memory::desc &diff_weights_layer_desc, const memory::desc &diff_weights_iter_desc, const memory::desc &diff_bias_desc, const memory::desc &diff_dst_layer_desc, const memory::desc &diff_dst_iter_desc, rnn_flags flags=rnn_flags::undef) | |
Initializes an GRU descriptor for backward propagation using prop_kind, direction, and memory descriptors. More... | |
Detailed Description
GRU descriptor for backward propagation.
Constructor & Destructor Documentation
◆ desc()
|
inline |
Initializes an GRU descriptor for backward propagation using prop_kind, direction, and memory descriptors.
flags is a parameter to the GRU descriptor and is currently ignored.
- Note
- All memory descriptors are allowed to be initialized with dnnl::memory::format_tag::any value of
format_kind.
src_iter_desc (simultaneously with diff_src_iter_desc), bias_desc (simultaneously with diff_bias_desc), and dst_iter_desc (simultaneously with diff_src_iter_desc) are allowed point to a zero memory descriptor, which would indicate that the GRU primitive should not use them and consider them to be zero values.
The documentation for this struct was generated from the following file:
- include/dnnl.hpp