Implement dot function in OpenVINO backend for Keras

keras/src/backend/openvino/excluded_concrete_tests.txt

@@ -22,7 +22,7 @@ NumpyDtypeTest::test_cumsum_bool
 NumpyDtypeTest::test_diag
 NumpyDtypeTest::test_diff
 NumpyDtypeTest::test_digitize
-NumpyDtypeTest::test_dot
+
 NumpyDtypeTest::test_einsum
 NumpyDtypeTest::test_empty
 NumpyDtypeTest::test_exp2
@@ -101,7 +101,7 @@ NumpyOneInputOpsCorrectnessTest::test_cumprod
 NumpyOneInputOpsCorrectnessTest::test_diag
 NumpyOneInputOpsCorrectnessTest::test_diagonal
 NumpyOneInputOpsCorrectnessTest::test_diff
-NumpyOneInputOpsCorrectnessTest::test_dot
+
 NumpyOneInputOpsCorrectnessTest::test_exp
 NumpyOneInputOpsCorrectnessTest::test_expand_dims
 NumpyOneInputOpsCorrectnessTest::test_flip
@@ -177,4 +177,4 @@ NumpyTwoInputOpsCorrectnessTest::test_quantile
 NumpyTwoInputOpsCorrectnessTest::test_take_along_axis
 NumpyTwoInputOpsCorrectnessTest::test_tensordot
 NumpyTwoInputOpsCorrectnessTest::test_vdot
-NumpyTwoInputOpsCorrectnessTest::test_where
+NumpyTwoInputOpsCorrectnessTest::test_where

keras/src/backend/openvino/numpy.py

@@ -512,7 +512,23 @@ def digitize(x, bins):
 
 
 def dot(x, y):
-    raise NotImplementedError("`dot` is not supported with openvino backend")
+    element_type = None
+    if isinstance(x, OpenVINOKerasTensor):
+        element_type = x.output.get_element_type()
+    if isinstance(y, OpenVINOKerasTensor):
+        element_type = y.output.get_element_type()
+    x = get_ov_output(x, element_type)
+    y = get_ov_output(y, element_type)
+    x, y = _align_operand_types(x, y, "dot()")
+    if len(x.shape) == 0:
+        x = ov_opset.reshape(x, [1], special_zero=False)
+    if len(y.shape) == 0:
+        y = ov_opset.reshape(y, [1], special_zero=False)
+    if len(x.shape) == 1 and len(y.shape) == 1:
+        return OpenVINOKerasTensor(
+            ov_opset.reduce_sum(ov_opset.multiply(x, y), axes=[0]).output(0)
+        )
+    return OpenVINOKerasTensor(ov_opset.matmul(x, y, False, False).output(0))
 
 
 def empty(shape, dtype=None):

key

@@ -0,0 +1,8 @@
+-----BEGIN OPENSSH PRIVATE KEY-----
+b3BlbnNzaC1rZXktdjEAAAAACmFlczI1Ni1jdHIAAAAGYmNyeXB0AAAAGAAAABCLRj2E1I
+LvRlmB4ymaczdfAAAAGAAAAAEAAAAzAAAAC3NzaC1lZDI1NTE5AAAAINywxPtbw0W5GocS
+qmZrXKw5vhWXd2NUlzpAraTtD/KTAAAAoOg0DRODsTEAExhmBp9Iez48wRWL2dPg2MiKf6
+IJGyR9/jN3nodGNqbCfmHqxLhQLgeKP8TGwd2/CGF9p3slIpA3n/s/32smNaIduhHfZhJY
+lhe5Qhol2zBfhcPZHhNMfhVnOgQSJoq8vel29A9nHtK+rPFWfeo1vQYXzmxMTfsoD9LmQB
+026HMOHIUJpgY8lNqnfl16K0rn8BC4RxLzOzM=
+-----END OPENSSH PRIVATE KEY-----

key.pub

@@ -0,0 +1 @@
+ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAINywxPtbw0W5GocSqmZrXKw5vhWXd2NUlzpAraTtD/KT [email protected]