ngraph::pass::SoftmaxFusion Class Reference

SoftmaxFusion transformation replaces following graph: More...

#include <softmax_fusion.hpp>

Inheritance diagram for ngraph::pass::SoftmaxFusion:

Data Fields
	NGRAPH_RTTI_DECLARATION

Detailed Description

SoftmaxFusion transformation replaces following graph:

       +---------------+
       │               │
       │     input     │
       │               │
       +---------------+
           │      │
           │      v
           │ +-----------+
           │ │           │
           │ │ ReduceMax │
           │ │           │
           │ +-----------+
           │      │
           │      │
           v      v
       +---------------+
       │               │
       │      Sub      │
       │               │
       +---------------+
               |
               |
               v
       +---------------+
       │               │
       │      Exp      │
       │               │
       +---------------+
           │      │
           │      v
           │ +-----------+
           │ │           │
           │ │ ReduceSum │
           │ │           │
           │ +-----------+
           │      │
           │      │
           v      v
        +-------------+
        |             │
        |     Div     │
        │             │
        +-------------+

to a single Softmax node

• Restrictions:
  • ReduceMax and ReduceSum axes must be scalar constants and they have to point to the same axis

---

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

• /home/jenkins/agent/workspace/openvino/documentation/upload_documentation/ovino_doc/build/_tmp_docs/openvino/inference-engine/src/transformations/include/transformations/common_optimizations/softmax_fusion.hpp