Descriptor for an elementwise forward propagation primitive. More...
#include <dnnl.hpp>
Public Member Functions | |
desc (prop_kind aprop_kind, algorithm aalgorithm, const memory::desc &data_desc, float alpha=0, float beta=0) | |
Constructs a descriptor for an elementwise forward propagation primitive. More... | |
Descriptor for an elementwise forward propagation primitive.
|
inline |
Constructs a descriptor for an elementwise forward propagation primitive.
aprop_kind | Propagation kind. Possible values are dnnl::prop_kind::forward_training, and dnnl::prop_kind::forward_inference. |
aalgorithm | Elementwise algorithm kind. |
data_desc | Source and destination memory descriptors. |
alpha | The alpha parameter for the elementwise operation. Specific meaning depends on the algorithm. |
beta | The beta parameter for the elementwise operation. Specific meaning depends on the algorithm. |