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

Descriptor for batch normalization forward propagation. More...

#include <mkldnn.hpp>

Collaboration diagram for mkldnn::batch_normalization_forward::desc:
Collaboration graph
[legend]

Public Member Functions

 desc (prop_kind aprop_kind, const memory::desc &src_desc, float epsilon, normalization_flags flags)
 Initializes a batch normalization descriptor for forward propagation using prop_kind (possible values are mkldnn::forward_training and mkldnn::forward_inference), memory descriptor data_desc, normalization parameter epsilon, and flags set using bit flags of type mkldnn_batch_normalization_desc_t. More...
 

Detailed Description

Descriptor for batch normalization forward propagation.

Constructor & Destructor Documentation

◆ desc()

mkldnn::batch_normalization_forward::desc::desc ( prop_kind  aprop_kind,
const memory::desc src_desc,
float  epsilon,
normalization_flags  flags 
)
inline

Initializes a batch normalization descriptor for forward propagation using prop_kind (possible values are mkldnn::forward_training and mkldnn::forward_inference), memory descriptor data_desc, normalization parameter epsilon, and flags set using bit flags of type mkldnn_batch_normalization_desc_t.

Note
In-place operation is supported; that is, dst points to the same memory as src.

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