A primitive to perform local response normalization (LRN) across or within channels. More...
Functions | |
| mkldnn_status_t MKLDNN_API | mkldnn_lrn_forward_desc_init (mkldnn_lrn_desc_t *lrn_desc, mkldnn_prop_kind_t prop_kind, mkldnn_alg_kind_t alg_kind, const mkldnn_memory_desc_t *data_desc, int local_size, double alpha, double beta, double k) |
Initializes an lrn_desc for forward propagation using prop_kind (possible values are mkldnn_forward_training or mkldnn_forward_inference), alg_kind, memory descriptor data_desc, and regularization parameters local_size, alpha, beta, and k. More... | |
| mkldnn_status_t MKLDNN_API | mkldnn_lrn_backward_desc_init (mkldnn_lrn_desc_t *lrn_desc, mkldnn_alg_kind_t alg_kind, const mkldnn_memory_desc_t *diff_data_desc, const mkldnn_memory_desc_t *data_desc, int local_size, double alpha, double beta, double k) |
Initializes an lrn_desc for backward propagation using alg_kind, memory descriptors data_desc, and diff_data_desc, and regularization parameters local_size, alpha, beta, and k. More... | |
Detailed Description
A primitive to perform local response normalization (LRN) across or within channels.
LRN accross channels:
![\[dst[n][c][h][w] = \left\{k + \frac{\alpha}{n_{l}} \sum\limits_{i=-(n_{l}-1)/2}^{(n_{l}+1)/2} (src[n][c+i][h][w])^2\right\}^{-\beta} src[n][c][h][w],\]](form_12.png)
LRN within channels:
![\[dst[n][c][h][w] = \left\{k + \frac{\alpha}{n_{l}} \sum\limits_{i=-(n_{l}-1)/2}^{(n_{l}+1)/2} (src[n][c][h+i][w+i])^2\right\}^{-\beta} src[n][c][h][w],\]](form_13.png)
where 
is the local_size.
Function Documentation
◆ mkldnn_lrn_backward_desc_init()
| mkldnn_status_t MKLDNN_API mkldnn_lrn_backward_desc_init | ( | mkldnn_lrn_desc_t * | lrn_desc, |
| mkldnn_alg_kind_t | alg_kind, | ||
| const mkldnn_memory_desc_t * | diff_data_desc, | ||
| const mkldnn_memory_desc_t * | data_desc, | ||
| int | local_size, | ||
| double | alpha, | ||
| double | beta, | ||
| double | k | ||
| ) |
Initializes an lrn_desc for backward propagation using alg_kind, memory descriptors data_desc, and diff_data_desc, and regularization parameters local_size, alpha, beta, and k.
◆ mkldnn_lrn_forward_desc_init()
| mkldnn_status_t MKLDNN_API mkldnn_lrn_forward_desc_init | ( | mkldnn_lrn_desc_t * | lrn_desc, |
| mkldnn_prop_kind_t | prop_kind, | ||
| mkldnn_alg_kind_t | alg_kind, | ||
| const mkldnn_memory_desc_t * | data_desc, | ||
| int | local_size, | ||
| double | alpha, | ||
| double | beta, | ||
| double | k | ||
| ) |
Initializes an lrn_desc for forward propagation using prop_kind (possible values are mkldnn_forward_training or mkldnn_forward_inference), alg_kind, memory descriptor data_desc, and regularization parameters local_size, alpha, beta, and k.
