.. index:: pair: struct; dnnl::softmax_v2_forward::desc .. _doxid-structdnnl_1_1softmax__v2__forward_1_1desc: struct dnnl::softmax_v2_forward::desc ===================================== .. toctree:: :hidden: Overview ~~~~~~~~ Descriptor for a softmax forward propagation primitive. :ref:`More...` .. ref-code-block:: cpp :class: doxyrest-overview-code-block #include struct desc { // fields :ref:`dnnl_softmax_v2_desc_t` :target:`data`; // construction :ref:`desc`(); :ref:`desc`( :ref:`prop_kind` aprop_kind, :ref:`algorithm` aalgorithm, const :ref:`memory::desc`& src_desc, const :ref:`memory::desc`& dst_desc, int softmax_axis ); }; .. _details-structdnnl_1_1softmax__v2__forward_1_1desc: Detailed Documentation ~~~~~~~~~~~~~~~~~~~~~~ Descriptor for a softmax forward propagation primitive. Construction ------------ .. index:: pair: function; desc .. _doxid-structdnnl_1_1softmax__v2__forward_1_1desc_1a3392feac2284301408f9b34d30d6d29c: .. ref-code-block:: cpp :class: doxyrest-title-code-block desc() Default constructor. Produces an empty object. .. index:: pair: function; desc .. _doxid-structdnnl_1_1softmax__v2__forward_1_1desc_1ae3c91ee70923fc43c6b83ce557bfa033: .. ref-code-block:: cpp :class: doxyrest-title-code-block desc( :ref:`prop_kind` aprop_kind, :ref:`algorithm` aalgorithm, const :ref:`memory::desc`& src_desc, const :ref:`memory::desc`& dst_desc, int softmax_axis ) Constructs a descriptor for a softmax forward propagation primitive. .. rubric:: Parameters: .. list-table:: :widths: 20 80 * - aprop_kind - Propagation kind. Possible values are :ref:`dnnl::prop_kind::forward_training `, and :ref:`dnnl::prop_kind::forward_inference `. * - aalgorithm - Softmax algorithm kind: either :ref:`dnnl::algorithm::softmax_accurate `, or :ref:`dnnl::algorithm::softmax_log `. * - src_desc - Source memory descriptor. * - dst_desc - Destination memory descriptor. * - softmax_axis - Axis over which softmax is computed.