layer_normalization_forward(layer_normalization_forward.primitive_desc pd)
Constructs a layer normalization forward propagation primitive.
|
primitive_desc(layer_normalization_backward.desc adesc,
engine aengine,
layer_normalization_forward.primitive_desc hint_fwd_pd) |
primitive_desc(layer_normalization_backward.desc adesc,
engine aengine,
layer_normalization_forward.primitive_desc hint_fwd_pd,
boolean allow_empty)
Constructs a primitive descriptor for a layer normalization backward
propagation primitive.
|
primitive_desc(layer_normalization_backward.desc adesc,
primitive_attr attr,
engine aengine,
layer_normalization_forward.primitive_desc hint_fwd_pd) |
primitive_desc(layer_normalization_backward.desc adesc,
primitive_attr attr,
engine aengine,
layer_normalization_forward.primitive_desc hint_fwd_pd,
boolean allow_empty)
Constructs a primitive descriptor for a layer normalization backward
propagation primitive.
|