Descriptor for a batch normalization forward propagation primitive. More...
#include <dnnl.hpp>
Public Member Functions | |
desc (prop_kind prop_kind, const memory::desc &data_desc, float epsilon, normalization_flags flags) | |
Constructs a batch normalization descriptor for forward propagation. More... | |
Descriptor for a batch normalization forward propagation primitive.
|
inline |
Constructs a batch normalization descriptor for forward propagation.
Inputs:
flags
flags
flags
Outputs:
flags
prop_kind
= dnnl_forward_trainingflags
and prop_kind
= dnnl_forward_trainingflags
and prop_kind
= dnnl_forward_training; must be queried for using dnnl_primitive_desc_query_md() after a corresponding primitive descriptor is createdprop_kind | Propagation kind. Possible values are dnnl_forward_training and dnnl_forward_inference. |
data_desc | Source and destination memory descriptors. |
epsilon | Batch normalization epsilon parameter. |
flags | Batch normalization flags (dnnl::normalization_flags). |