dnnl::lstm_forward::desc Struct Reference

Descriptor for LSTM forward propagation. More...

#include <dnnl.hpp>

Collaboration diagram for dnnl::lstm_forward::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, 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 rnn_training_f32.cpp.

Constructor & Destructor Documentation

desc()

dnnl::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 dnnl::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 dnnl::memory::format_tag::any value of format_kind.

---

The documentation for this struct was generated from the following file:

• include/dnnl.hpp