LSTM descriptor for backward propagation. More...
#include <dnnl.hpp>
Collaboration diagram for dnnl::lstm_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 &src_iter_c_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 &dst_iter_c_desc, const memory::desc &diff_src_layer_desc, const memory::desc &diff_src_iter_desc, const memory::desc &diff_src_iter_c_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, const memory::desc &diff_dst_iter_c_desc, rnn_flags flags=rnn_flags::undef) | |
Initializes an LSTM descriptor for backward propagation using prop_kind, direction, and memory descriptors. More... | |
Detailed Description
LSTM descriptor for backward propagation.
- Examples
- rnn_training_f32.cpp.
Constructor & Destructor Documentation
◆ desc()
|
inline |
Initializes an LSTM descriptor for backward propagation using prop_kind, direction, and memory descriptors.
flags is a parameter to the LSTM 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), src_iter_c_desc (simultaneously with diff_src_iter_c_desc), bias_desc (simultaneously with diff_bias_desc), dst_iter_desc (simultaneously with diff_src_iter_desc) and dst_iter_c_desc (simultaneously with diff_src_iter_c_desc) are allowed point to a zero memory descriptor, which would indicate that the LSTM 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