Intel(R) Math Kernel Library for Deep Neural Networks (Intel(R) MKL-DNN)  1.0.4
Performance library for Deep Learning
Public Member Functions | List of all members
mkldnn::lstm_forward::desc Struct Reference

Descriptor for LSTM forward propagation. More...

#include <mkldnn.hpp>

Collaboration diagram for mkldnn::lstm_forward::desc:
Collaboration graph
[legend]

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, rnn_flags flags=rnn_flags::undef)
 Initializes an LSTM descriptor for forward propagation using prop_kind, direction, and memory descriptors. More...
 

Detailed Description

Descriptor for LSTM forward propagation.

Examples:
cpu_rnn_inference_f32.cpp, cpu_rnn_inference_int8.cpp, and cpu_rnn_training_f32.cpp.

Constructor & Destructor Documentation

◆ desc()

mkldnn::lstm_forward::desc::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,
rnn_flags  flags = rnn_flags::undef 
)
inline

Initializes an LSTM descriptor for forward propagation using prop_kind, direction, and memory descriptors.

Note
If prop_kind equals mkldnn::forward_training, you must query a workspace memory descriptor before creating the primitive.

flags is a parameter to the LSTM descriptor and is currently ignored.

src_iter_desc, src_iter_c_desc, bias_desc, dst_iter_desc and dst_iter_c_desc are allowed to point to a zero memory descriptor, which would indicate that the LSTM primitive should not use them.

Note
All memory descriptors except src_iter_desc can be initialized with an mkldnn::memory::format_tag::any value of format_kind.
The documentation for this struct was generated from the following file: