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_backward::desc Struct Reference

LSTM descriptor for backward propagation. More...

#include <mkldnn.hpp>

Collaboration diagram for mkldnn::lstm_backward::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, 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:
cpu_rnn_training_f32.cpp.

Constructor & Destructor Documentation

◆ desc()

mkldnn::lstm_backward::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,
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 
)
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 mkldnn::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: