Add description to mish fuse transformation

openvinotoolkit · Aug 3, 2020 · b42ba36 · b42ba36
1 parent 37b0f00
commit b42ba36
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/inference-engine/src/transformations/include/transformations/mish_fusion.hpp b/inference-engine/src/transformations/include/transformations/mish_fusion.hpp
@@ -21,6 +21,11 @@ class TRANSFORMATIONS_API MishFusion;
 }  // namespace pass
 }  // namespace ngraph
 
+/**
+ * @ingroup ie_transformation_common_api
+ * @brief MishFusion transformation replaces group of
+ * operations: x * tanh(log(exp(x) + 1)) to Mish op.
+ */
 class ngraph::pass::MishFusion: public ngraph::pass::MatcherPass {
 public:
     MishFusion();