[OpenVINO backend] Update ndim behavior

keras/src/backend/openvino/core.py

@@ -3,11 +3,11 @@
 
 import numpy as np
 import openvino as ov
-import openvino.runtime.opset14 as ov_opset
+import openvino.opset14 as ov_opset
 from openvino import Model
 from openvino import Tensor
 from openvino import compile_model
-from openvino.runtime import Type
+from openvino import Type
 
 from keras.src import tree
 from keras.src.backend.common import KerasVariable

keras/src/backend/openvino/math.py

@@ -1,4 +1,4 @@
-import openvino.runtime.opset14 as ov_opset
+import openvino.opset14 as ov_opset
 from openvino import Type
 
 from keras.src.backend.openvino.core import OpenVINOKerasTensor

keras/src/backend/openvino/nn.py

@@ -1,4 +1,4 @@
-import openvino.runtime.opset14 as ov_opset
+import openvino.opset14 as ov_opset
 from openvino import Type
 
 from keras.src import backend

keras/src/backend/openvino/numpy.py

@@ -1,5 +1,5 @@
 import numpy as np
-import openvino.runtime.opset14 as ov_opset
+import openvino.opset14 as ov_opset
 from openvino import Type
 
 from keras.src.backend import config
@@ -1167,9 +1167,7 @@ def nan_to_num(x, nan=0.0, posinf=None, neginf=None):
 
 def ndim(x):
     x = get_ov_output(x)
-    x_shape = ov_opset.shape_of(x).output(0)
-    x_dim = ov_opset.shape_of(x_shape, "i64")
-    return x_dim
+    return len(x.get_partial_shape().to_shape())
 
 
 def nonzero(x):

keras/src/backend/openvino/random.py

@@ -1,5 +1,5 @@
 import numpy as np
-import openvino.runtime.opset14 as ov_opset
+import openvino.opset14 as ov_opset
 from openvino import Type
 
 from keras.src.backend.config import floatx

keras/src/backend/openvino/trainer.py

@@ -1,6 +1,6 @@
 import numpy as np
 import openvino as ov
-import openvino.runtime.opset14 as ov_opset
+import openvino.opset14 as ov_opset
 
 from keras.src import backend
 from keras.src import callbacks as callbacks_module