Descriptor for a batch normalization forward propagation primitive. More...
#include <dnnl.hpp>

Public Member Functions | |
| desc (prop_kind aprop_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.
| aprop_kind | Propagation kind. Possible values are dnnl::prop_kind::forward_training and dnnl::prop_kind::forward_inference. |
| data_desc | Source and destination memory descriptors. |
| epsilon | Batch normalization epsilon parameter. |
| flags | Batch normalization flags (dnnl::normalization_flags). |