oneAPI Deep Neural Network Library (oneDNN)
Performance library for Deep Learning
1.96.0
dnnl::prelu_forward::desc Struct Reference

Descriptor for a PReLU forward propagation primitive. More...

#include <dnnl.hpp>

Public Member Functions

 desc (prop_kind aprop_kind, const memory::desc &data_desc, const memory::desc &weight_desc)
 Constructs a descriptor for a PReLU forward propagation primitive. More...
 

Detailed Description

Descriptor for a PReLU forward propagation primitive.

Examples:
prelu.cpp.

Constructor & Destructor Documentation

◆ desc()

dnnl::prelu_forward::desc::desc ( prop_kind  aprop_kind,
const memory::desc data_desc,
const memory::desc weight_desc 
)
inline

Constructs a descriptor for a PReLU forward propagation primitive.

Parameters
aprop_kindPropagation kind. Possible values are dnnl::prop_kind::forward_training, and dnnl::prop_kind::forward_inference.
data_descSource and destination memory descriptors.
weight_descAlpha parameters memory descriptor.

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