Publishing 2019 R3 content

[platform/upstream/dldt.git] / model-optimizer / mo / front / common / partial_infer / eltwise.py

diff --git a/model-optimizer/mo/front/common/partial_infer/eltwise.py b/model-optimizer/mo/front/common/partial_infer/eltwise.py
index 12d4b80..7cfdb15 100644 (file)
--- a/model-optimizer/mo/front/common/partial_infer/eltwise.py
+++ b/model-optimizer/mo/front/common/partial_infer/eltwise.py
@@ -14,9 +14,8 @@
  limitations under the License.
 """
 
-import numpy as np
-import logging as log
 import networkx as nx
+import numpy as np
 
 from mo.front.common.partial_infer.utils import int64_array
 from mo.graph.graph import Node
@@ -90,3 +89,10 @@ def eltwise_infer(node, op=None, **kwargs):
         node.out_node().value = values[0]
         for i in range(len(values) - 1):
             node.out_node().value = op(node.out_node().value, values[i + 1])
+
+
+def bias_add_infer(node, op):
+    if node.in_port(0).data.get_value() is not None and node.in_port(1).data.get_value() is not None and op is not None:
+        node.out_port(0).data.set_value(op(node.in_port(0).data.get_value(), node.in_port(1).data.get_value()))
+    else:
+        node.out_port(0).data.set_shape(node.in_port(0).data.get_shape())