IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

BitwiseElementwise ops init in MO (#20386)

Browse Source
This commit is contained in:
Katarzyna Mitrus 2023-10-12 18:18:34 +02:00 committed by GitHub
parent 6519afd4d3
commit 29475c738e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 71 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
tools/mo/openvino/tools/mo/ops/elementwise.py
View File

@ -246,3 +246,27 @@ class Sqrt(UnaryElementwise):
if np.issubdtype(a.dtype, np.signedinteger):
return float32_array(a.astype(np.float32) ** 0.5)
return a ** 0.5
class BitwiseAnd(Elementwise):
op = 'BitwiseAnd'
op_type = 'BitwiseAnd'
version = 'opset13'
class BitwiseOr(Elementwise):
op = 'BitwiseOr'
op_type = 'BitwiseOr'
version = 'opset13'
class BitwiseXor(Elementwise):
op = 'BitwiseXor'
op_type = 'BitwiseXor'
version = 'opset13'
class BitwiseNot(UnaryElementwise):
op = 'BitwiseNot'
op_type = 'BitwiseNot'
version = 'opset13'

47
tools/mo/unit_tests/mo/utils/ir_reader/ops_test.py
View File

@ -6,6 +6,7 @@ import tempfile
import numpy as np
from pathlib import Path
import openvino.runtime.opset13 as opset13
import openvino.runtime.opset12 as opset12
import openvino.runtime.opset11 as opset11
import openvino.runtime.opset10 as opset10
@ -245,3 +246,49 @@ class TestOps(unittest.TestCase):
self.assertEqual(gn_node["version"], "opset12")
self.assertEqual(gn_node['num_groups'], 1)
self.assertEqual(gn_node['epsilon'], 1e-06)
def test_bitwise_and_13(self):
a = opset13.parameter([4, 1], name="A", dtype=np.int32)
b = opset13.parameter([1, 2], name="B", dtype=np.int32)
op = opset13.bitwise_and(a, b)
model = Model(op, [a, b])
graph = TestOps.check_graph_can_save(model, "bitwise_and_model")
op_node = graph.get_op_nodes(op="BitwiseAnd")[0]
self.assertListEqual(op_node.out_port(0).data.get_shape().tolist(), [4, 2])
self.assertEqual(op_node["version"], "opset13")
self.assertEqual(op_node["auto_broadcast"], "numpy")
def test_bitwise_or_13(self):
a = opset13.parameter([4, 1], name="A", dtype=np.int32)
b = opset13.parameter([1, 2], name="B", dtype=np.int32)
op = opset13.bitwise_or(a, b)
model = Model(op, [a, b])
graph = TestOps.check_graph_can_save(model, "bitwise_or_model")
op_node = graph.get_op_nodes(op="BitwiseOr")[0]
self.assertListEqual(op_node.out_port(0).data.get_shape().tolist(), [4, 2])
self.assertEqual(op_node["version"], "opset13")
self.assertEqual(op_node["auto_broadcast"], "numpy")
def test_bitwise_xor_13(self):
a = opset13.parameter([4, 1], name="A", dtype=np.int32)
b = opset13.parameter([1, 2], name="B", dtype=np.int32)
op = opset13.bitwise_xor(a, b)
model = Model(op, [a, b])
graph = TestOps.check_graph_can_save(model, "bitwise_xor_model")
op_node = graph.get_op_nodes(op="BitwiseXor")[0]
self.assertListEqual(op_node.out_port(0).data.get_shape().tolist(), [4, 2])
self.assertEqual(op_node["version"], "opset13")
self.assertEqual(op_node["auto_broadcast"], "numpy")
def test_bitwise_not_13(self):
a = opset13.parameter([4, 2], name="A", dtype=np.int32)
op = opset13.bitwise_not(a)
model = Model(op, [a])
graph = TestOps.check_graph_can_save(model, "bitwise_not_model")
op_node = graph.get_op_nodes(op="BitwiseNot")[0]
self.assertListEqual(op_node.out_port(0).data.get_shape().tolist(), [4, 2])
self.assertEqual(op_node["version"], "opset13")