Descriptor for layer normalization forward propagation. More...
#include <dnnl.hpp>
Public Member Functions | |
desc (prop_kind aprop_kind, const memory::desc &src_desc, const memory::desc &stat_desc, float epsilon, normalization_flags flags) | |
Initializes a layer normalization descriptor for forward propagation using prop_kind (possible values are dnnl::forward_training and dnnl::forward_inference), memory descriptor data_desc , normalization parameter epsilon , and flags set using bit flags of type dnnl_layer_normalization_desc_t. More... | |
Descriptor for layer normalization forward propagation.
|
inline |
Initializes a layer normalization descriptor for forward propagation using prop_kind
(possible values are dnnl::forward_training and dnnl::forward_inference), memory descriptor data_desc
, normalization parameter epsilon
, and flags
set using bit flags of type dnnl_layer_normalization_desc_t.