openvino
eb869158 - Make RMSFusion support ReduceMean with the last axis represented by a positive value (#33351)

Commit
65 days ago
Make RMSFusion support ReduceMean with the last axis represented by a positive value (#33351) ### Details: For RMSFusion, `mean_axes` only matches constants with a value of -1, but the callback allows two cases: -1 or rank-1. Therefore, if the graph uses "positive value representing the last axis" (i.e., rank-1), the pattern will not match at all and the callback will not have a chance to execute. As a result, last axis with a positive value will indeed not match. This PR fixes the issue by verifying that the ReduceMean has exactly one axis, and by checking the axis value is -1 or rank-1 in callback. ### Jira Ticket: - https://jira.devtools.intel.com/browse/CVS-178799 --------- Co-authored-by: Mikhail Ryzhov <mikhail.ryzhov@intel.com>
Author
Parents
Loading