Add hsigmoid fusing for MO (#2750)
* Add hsigmoid fusing for MO * Update Bom file * Remove comments * Refactoring hsigmoid fusion according to review * Add div and mul patterns for hsigmoid fusion * Refactoring code according to review * Fix HSigmoid fusion transformation
This commit is contained in:
1 parent
7bd8391aba
commit
ce19e1fb9a
4 files changed
+543
No files matched your search
@@ -248,6 +248,12 @@ class HSwish(Activation):
|
||||
operation = staticmethod(lambda x: x * np.minimum(np.maximum(x + 3.0, 0.0), 6.0) / 6.0)
|
||||
|
||||
|
||||
class HSigmoid(Activation):
|
||||
op = 'HSigmoid'
|
||||
version = 'opset5'
|
||||
operation = staticmethod(lambda x: np.minimum(np.maximum(x + 3.0, 0.0), 6.0) / 6.0)
|
||||
|
||||
|
||||
class Swish(Op):
|
||||
op = 'Swish'
|
||||
|
||||
|
||||
Reference in new issue
Block a user