mxnet
|
Classes | |
struct | dnnl_batch_normalization_desc_t |
A descriptor of a Batch Normalization operation. More... | |
Functions | |
dnnl_status_t DNNL_API | dnnl_batch_normalization_forward_desc_init (dnnl_batch_normalization_desc_t *bnrm_desc, dnnl_prop_kind_t prop_kind, const dnnl_memory_desc_t *data_desc, float epsilon, unsigned flags) |
dnnl_status_t DNNL_API | dnnl_batch_normalization_backward_desc_init (dnnl_batch_normalization_desc_t *bnrm_desc, dnnl_prop_kind_t prop_kind, const dnnl_memory_desc_t *diff_data_desc, const dnnl_memory_desc_t *data_desc, float epsilon, unsigned flags) |
dnnl_status_t DNNL_API dnnl_batch_normalization_backward_desc_init | ( | dnnl_batch_normalization_desc_t * | bnrm_desc, |
dnnl_prop_kind_t | prop_kind, | ||
const dnnl_memory_desc_t * | diff_data_desc, | ||
const dnnl_memory_desc_t * | data_desc, | ||
float | epsilon, | ||
unsigned | flags | ||
) |
Initializes a descriptor for a batch normalization backward propagation primitive.
bnrm_desc | Output descriptor for batch normalization primitive. |
prop_kind | Propagation kind. Possible values are dnnl_backward_data and dnnl_backward (diffs for all parameters are computed in this case). |
diff_data_desc | Diff source and diff destination memory descriptor. |
data_desc | Source memory descriptor. |
epsilon | Batch normalization epsilon parameter. |
flags | Batch normalization flags (dnnl_normalization_flags_t). |
dnnl_status_t DNNL_API dnnl_batch_normalization_forward_desc_init | ( | dnnl_batch_normalization_desc_t * | bnrm_desc, |
dnnl_prop_kind_t | prop_kind, | ||
const dnnl_memory_desc_t * | data_desc, | ||
float | epsilon, | ||
unsigned | flags | ||
) |
Initializes a descriptor for a batch normalization forward propagation primitive.
bnrm_desc | Output descriptor for batch normalization primitive. |
prop_kind | Propagation kind. Possible values are dnnl_forward_training and dnnl_forward_inference. |
data_desc | Source and destination memory descriptor. |
epsilon | Batch normalization epsilon parameter. |
flags | Batch normalization flags (dnnl_normalization_flags_t). |