.. index:: pair: struct; dnnl_softmax_v2_desc_t .. _doxid-structdnnl__softmax__v2__desc__t: struct dnnl_softmax_v2_desc_t ============================= .. toctree:: :hidden: Overview ~~~~~~~~ A descriptor of a Softmax operation. :ref:`More...` .. ref-code-block:: cpp :class: doxyrest-overview-code-block #include struct dnnl_softmax_v2_desc_t { // fields :ref:`dnnl_primitive_kind_t` :ref:`primitive_kind`; :ref:`dnnl_prop_kind_t` :ref:`prop_kind`; :ref:`dnnl_memory_desc_t` :ref:`src_desc`; :ref:`dnnl_memory_desc_t` :ref:`diff_src_desc`; int :ref:`softmax_axis`; :ref:`dnnl_alg_kind_t` :ref:`alg_kind`; :ref:`dnnl_memory_desc_t` :ref:`dst_desc`; :ref:`dnnl_memory_desc_t` :ref:`diff_dst_desc`; }; .. _details-structdnnl__softmax__v2__desc__t: Detailed Documentation ~~~~~~~~~~~~~~~~~~~~~~ A descriptor of a Softmax operation. Fields ------ .. index:: pair: variable; primitive_kind .. _doxid-structdnnl__softmax__v2__desc__t_1aa48832e3b4daad7ad378329e028b8f04: .. ref-code-block:: cpp :class: doxyrest-title-code-block :ref:`dnnl_primitive_kind_t` primitive_kind The kind of primitive. Used for self-identifying the primitive descriptor. Must be :ref:`dnnl_softmax_v2 `. .. index:: pair: variable; prop_kind .. _doxid-structdnnl__softmax__v2__desc__t_1a4092a09322b6b5c4bc74cbc33a23d5a9: .. ref-code-block:: cpp :class: doxyrest-title-code-block :ref:`dnnl_prop_kind_t` prop_kind The kind of propagation. Possible values: :ref:`dnnl_forward_training `, :ref:`dnnl_forward_inference `, and :ref:`dnnl_backward_data `. .. index:: pair: variable; src_desc .. _doxid-structdnnl__softmax__v2__desc__t_1a3a34370e73b48cdecf3998e62b0d2c46: .. ref-code-block:: cpp :class: doxyrest-title-code-block :ref:`dnnl_memory_desc_t` src_desc Source memory descriptor. .. index:: pair: variable; diff_src_desc .. _doxid-structdnnl__softmax__v2__desc__t_1a8cc0df0b19a0c9a6a03e5a6249d27df6: .. ref-code-block:: cpp :class: doxyrest-title-code-block :ref:`dnnl_memory_desc_t` diff_src_desc Source gradient memory descriptor. .. index:: pair: variable; softmax_axis .. _doxid-structdnnl__softmax__v2__desc__t_1a184017535be31d3d6d133ac218e236bb: .. ref-code-block:: cpp :class: doxyrest-title-code-block int softmax_axis The axis along which to perform the softmax. .. index:: pair: variable; alg_kind .. _doxid-structdnnl__softmax__v2__desc__t_1ae95479aa51f590448e0c02a93149277e: .. ref-code-block:: cpp :class: doxyrest-title-code-block :ref:`dnnl_alg_kind_t` alg_kind Softmax algorithm. Possible values: :ref:`dnnl_softmax_accurate ` and :ref:`dnnl_softmax_log `. .. index:: pair: variable; dst_desc .. _doxid-structdnnl__softmax__v2__desc__t_1ac88abe94c32e3a054a6c98d84efcc966: .. ref-code-block:: cpp :class: doxyrest-title-code-block :ref:`dnnl_memory_desc_t` dst_desc Destination memory descriptor. .. index:: pair: variable; diff_dst_desc .. _doxid-structdnnl__softmax__v2__desc__t_1ab3c01315174f169de709851ba18ac1c5: .. ref-code-block:: cpp :class: doxyrest-title-code-block :ref:`dnnl_memory_desc_t` diff_dst_desc Destination gradient memory descriptor.