| Package | Description |
|---|---|
| org.bytedeco.dnnl | |
| org.bytedeco.dnnl.global |
| Modifier and Type | Method and Description |
|---|---|
layer_normalization_backward.desc |
layer_normalization_backward.desc.data(dnnl_layer_normalization_v2_desc_t setter) |
layer_normalization_forward.desc |
layer_normalization_forward.desc.data(dnnl_layer_normalization_v2_desc_t setter) |
| Modifier and Type | Method and Description |
|---|---|
static int |
dnnl.dnnl_layer_normalization_v2_backward_desc_init(dnnl_layer_normalization_v2_desc_t lnrm_desc,
int prop_kind,
dnnl_memory_desc_t diff_src_desc,
dnnl_memory_desc_t diff_dst_desc,
dnnl_memory_desc_t src_desc,
dnnl_memory_desc_t stat_desc,
float epsilon,
int flags)
Initializes a descriptor for a layer normalization v2 backward propagation
primitive.
|
static int |
dnnl.dnnl_layer_normalization_v2_forward_desc_init(dnnl_layer_normalization_v2_desc_t lnrm_desc,
int prop_kind,
dnnl_memory_desc_t src_desc,
dnnl_memory_desc_t dst_desc,
dnnl_memory_desc_t stat_desc,
float epsilon,
int flags)
\} dnnl_api_layer_normalization
|
Copyright © 2022. All rights reserved.