MO clean from IR v7 and other legacy code (#1521)
* Remove unnnecessary ir_version checks in the MO * Cleaned up 'backend_attrs_v2' function * Small clean up from the 'TFCustomSubgraphCall' * Clean up the MO extractor attributes mapping * Renamed PreluOp to PReLU
This commit is contained in:
1 parent
aef6016298
commit
dec7df17ed
27 files changed
+125
-656
No files matched your search
@@ -16,7 +16,7 @@
|
||||
import numpy as np
|
||||
|
||||
from extensions.back.ForceStrictPrecision import ForceStrictPrecision
|
||||
from extensions.ops.prelu import PreluOp
|
||||
from extensions.ops.prelu import PReLU
|
||||
from mo.back.replacement import BackReplacementPattern
|
||||
from mo.graph.graph import Graph, rename_node
|
||||
from mo.ops.const import Const
|
||||
@@ -45,7 +45,7 @@ class LeakyReLUMutation(BackReplacementPattern):
|
||||
|
||||
rename_node(relu, relu_name + '/to_delete')
|
||||
# Create PReLU op and reconnect input/output from LeakyReLU to PReLU
|
||||
prelu = PreluOp(graph, dict(name=relu_name)).create_node()
|
||||
prelu = PReLU(graph, dict(name=relu_name)).create_node()
|
||||
rename_node(prelu, relu_name)
|
||||
|
||||
const = Const(graph, dict(name=relu_name + "/weights", value=np.array([relu.negative_slope]))).create_node()
|
||||
|
||||
Reference in new issue
Block a user