keras-nightly 3.14.0.dev2026020304__py3-none-any.whl → 3.14.0.dev2026020404__py3-none-any.whl

keras/src/backend/openvino/numpy.py

@@ -802,7 +802,78 @@ def count_nonzero(x, axis=None):
 
 
 def cross(x1, x2, axisa=-1, axisb=-1, axisc=-1, axis=None):
-    raise NotImplementedError("`cross` is not supported with openvino backend")
+    if axis is not None:
+        axisa = axisb = axisc = axis
+
+    x1 = get_ov_output(x1)
+    x2 = get_ov_output(x2)
+
+    x1, x2 = _align_operand_types(x1, x2, "cross()")
+
+    shape1 = x1.get_partial_shape()
+    shape2 = x2.get_partial_shape()
+
+    # Rank Normalization
+    rank1 = shape1.rank.get_length()
+    rank2 = shape2.rank.get_length()
+
+    axisa = canonicalize_axis(axisa, rank1)
+    axisb = canonicalize_axis(axisb, rank2)
+    axisc = canonicalize_axis(axisc, rank1 if rank1 > rank2 else rank2)
+
+    d1 = shape1[axisa].get_length()
+    d2 = shape2[axisb].get_length()
+
+    if d1 not in (2, 3) or d2 not in (2, 3):
+        raise ValueError(
+            "Dimension of vectors for cross product must be 2 or 3. "
+            f"Got dimensions {d1} and {d2} for inputs x1 and x2."
+        )
+
+    # Pad to 3D by adding a zero component.
+    def pad_to_3d(x, dim, ax):
+        if dim == 3:
+            return x
+
+        # Create a slice of zeros with the same type as x
+        slice0 = ov_opset.gather(
+            x,
+            ov_opset.constant([0], Type.i32),
+            ov_opset.constant(ax, Type.i32),
+        )
+        zeros = ov_opset.multiply(
+            slice0,
+            ov_opset.constant(0, x.get_element_type()),
+        )
+
+        return ov_opset.concat([x, zeros], ax)
+
+    x1_3d = pad_to_3d(x1, d1, axisa)
+    x2_3d = pad_to_3d(x2, d2, axisb)
+
+    # Split Vectors
+    u = ov_opset.split(x1_3d, ov_opset.constant(axisa, Type.i32), 3).outputs()
+    v = ov_opset.split(x2_3d, ov_opset.constant(axisb, Type.i32), 3).outputs()
+
+    # u x v = (u2*v3 - u3*v2, u3*v1 - u1*v3, u1*v2 - u2*v1)
+    res_x = ov_opset.subtract(
+        ov_opset.multiply(u[1], v[2]), ov_opset.multiply(u[2], v[1])
+    )
+    res_y = ov_opset.subtract(
+        ov_opset.multiply(u[2], v[0]), ov_opset.multiply(u[0], v[2])
+    )
+    res_z = ov_opset.subtract(
+        ov_opset.multiply(u[0], v[1]), ov_opset.multiply(u[1], v[0])
+    )
+
+    # If dim was 2D, we remove the padded zero component.
+    if d1 == 2 and d2 == 2:
+        result = res_z
+        result = ov_opset.squeeze(result, ov_opset.constant([axisc], Type.i32))
+    else:
+        result = ov_opset.concat([res_x, res_y, res_z], axisc)
+
+    return OpenVINOKerasTensor(result.output(0))
 
 
 def cumprod(x, axis=None, dtype=None):

keras/src/testing/__init__.py

@@ -2,6 +2,7 @@ from keras.src.testing.test_case import TestCase
 from keras.src.testing.test_case import jax_uses_gpu
 from keras.src.testing.test_case import jax_uses_tpu
 from keras.src.testing.test_case import tensorflow_uses_gpu
+from keras.src.testing.test_case import tensorflow_uses_tpu
 from keras.src.testing.test_case import torch_uses_gpu
 from keras.src.testing.test_case import uses_gpu
 from keras.src.testing.test_case import uses_tpu

keras/src/testing/test_case.py

@@ -8,7 +8,6 @@ import numpy as np
 from absl.testing import parameterized
 
 from keras.src import backend
-from keras.src import distribution
 from keras.src import ops
 from keras.src import tree
 from keras.src import utils
@@ -626,43 +625,68 @@ class TestCase(parameterized.TestCase, unittest.TestCase):
                         self.assertEqual(dtype, "float32")
 
 
-def tensorflow_uses_gpu():
-    return backend.backend() == "tensorflow" and uses_gpu()
+def _jax_uses(device_type):
+    import jax
 
+    return jax.default_backend() == device_type
 
-def jax_uses_gpu():
-    return backend.backend() == "jax" and uses_gpu()
 
+def _tensorflow_uses(device_type):
+    import tensorflow as tf
 
-def torch_uses_gpu():
-    if backend.backend() != "torch":
-        return False
-    from keras.src.backend.torch.core import get_device
+    return len(tf.config.list_physical_devices(device_type.upper())) > 0
+
+
+def _torch_uses(device_type):
+    if device_type == "gpu":
+        from keras.src.backend.torch.core import get_device
 
-    return get_device() == "cuda"
+        return get_device() == "cuda"
+    return device_type == "cpu"
 
 
 def uses_gpu():
-    # Condition used to skip tests when using the GPU
-    devices = distribution.list_devices()
-    if any(d.startswith("gpu") for d in devices):
-        return True
-    return False
+    if not hasattr(uses_gpu, "_value"):
+        if backend.backend() == "tensorflow":
+            uses_gpu._value = _tensorflow_uses("gpu")
+        elif backend.backend() == "jax":
+            uses_gpu._value = _jax_uses("gpu")
+        elif backend.backend() == "torch":
+            uses_gpu._value = _torch_uses("gpu")
+        else:
+            uses_gpu._value = False
+    return uses_gpu._value
+
+
+def uses_tpu():
+    if not hasattr(uses_tpu, "_value"):
+        if backend.backend() == "tensorflow":
+            uses_tpu._value = _tensorflow_uses("tpu")
+        elif backend.backend() == "jax":
+            uses_tpu._value = _jax_uses("tpu")
+        else:
+            uses_tpu._value = False
+    return uses_tpu._value
+
+
+def jax_uses_gpu():
+    return backend.backend() == "jax" and uses_gpu()
 
 
 def jax_uses_tpu():
     return backend.backend() == "jax" and uses_tpu()
 
 
-def uses_tpu():
-    # Condition used to skip tests when using the TPU
-    try:
-        devices = distribution.list_devices()
-        if any(d.startswith("tpu") for d in devices):
-            return True
-    except AttributeError:
-        return False
-    return False
+def tensorflow_uses_gpu():
+    return backend.backend() == "tensorflow" and uses_gpu()
+
+
+def tensorflow_uses_tpu():
+    return backend.backend() == "tensorflow" and uses_tpu()
+
+
+def torch_uses_gpu():
+    return backend.backend() == "torch" and uses_gpu()
 
 
 def create_keras_tensors(input_shape, dtype, sparse, ragged):

keras/src/version.py

@@ -1,7 +1,7 @@
 from keras.src.api_export import keras_export
 
 # Unique source of truth for the version number.
-__version__ = "3.14.0.dev2026020304"
+__version__ = "3.14.0.dev2026020404"
 
 
 @keras_export("keras.version")

{keras_nightly-3.14.0.dev2026020304.dist-info → keras_nightly-3.14.0.dev2026020404.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: keras-nightly
-Version: 3.14.0.dev2026020304
+Version: 3.14.0.dev2026020404
 Summary: Multi-backend Keras
 Author-email: Keras team <keras-users@googlegroups.com>
 License: Apache License 2.0

{keras_nightly-3.14.0.dev2026020304.dist-info → keras_nightly-3.14.0.dev2026020404.dist-info}/RECORD

@@ -128,7 +128,7 @@ keras/regularizers/__init__.py,sha256=542Shphw7W8h4Dyf2rmqMKUECVZ8IVBvN9g1LWhz-b
 keras/saving/__init__.py,sha256=KvL2GZxjvgFgEhvEnkvqjIR9JSNHKz-NWZacXajsjLI,1298
 keras/src/__init__.py,sha256=Gi4S7EiCMkE03PbdGNpFdaUYySWDs_FcAJ8Taz9Y1BE,684
 keras/src/api_export.py,sha256=gXOkBOnmscV013WAc75lc4Up01-Kkg9EylIAT_QWctg,1173
-keras/src/version.py,sha256=EgrYJtA7YKUOV4vpq9co7mADp2YlsvvhgZYUYS4YnbY,204
+keras/src/version.py,sha256=FhnIBR0HLjhX-AjhyRTy_cn4TF4qatX-aS3GrHpvUok,204
 keras/src/activations/__init__.py,sha256=0nL3IFDB9unlrMz8ninKOWo-uCHasTUpTo1tXZb2u44,4433
 keras/src/activations/activations.py,sha256=mogPggtp4CGldI3VOPNmesRxp6EbiR1_i4KLGaVwzL8,17614
 keras/src/applications/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -198,7 +198,7 @@ keras/src/backend/openvino/layer.py,sha256=5RdvaH1yOyPAphjKiuQAK1H_yZFYKE1Hp7c5b
 keras/src/backend/openvino/linalg.py,sha256=L6a4MFGND2wWzPVCh44cwuOgkcC4wJTo8Xy3HwW04lg,1614
 keras/src/backend/openvino/math.py,sha256=qw9kX2sJ2qr0dBJF12Ey0E2GcwixPUqoev6UcNra4NI,3944
 keras/src/backend/openvino/nn.py,sha256=zULPxdwVO7JDZUUtsuoEEPCLQ09ew8z8T6G_i_NEqrM,23741
-keras/src/backend/openvino/numpy.py,sha256=8FKiEmrGHEvhXhpE84j_ew87muFgcr9QMxNGNX5OUig,113533
+keras/src/backend/openvino/numpy.py,sha256=vPmpxizrgQyShUTeAaVSI38ANX6z2uHYiAf9b-4n9e4,115691
 keras/src/backend/openvino/random.py,sha256=4hRUtIP6qJxO3Qy9uH1x6jSuJna3nWPdUf4x2QU8-ew,5575
 keras/src/backend/openvino/rnn.py,sha256=ErmuZLPSgG9qU-NfYPPvBZ6Ysy8k-fA4g19Vhqq7OVQ,866
 keras/src/backend/openvino/trainer.py,sha256=bMmtSALqydqdS6ke-5sYW5fgxZDshDH810p_C0xCRTg,9087
@@ -550,8 +550,8 @@ keras/src/saving/orbax_util.py,sha256=ArJI9hQODUyyvzCiXt8AS3VH6E4SL0vF02-RHBk30g
 keras/src/saving/saving_api.py,sha256=X-zsTum57M3AWf_cYqhq_o5wgLcFxNh1ditoxbL_0LY,14694
 keras/src/saving/saving_lib.py,sha256=bRI8TeNOlflTfX3njSkkwNv-VYip-OW7ienIm0lL96I,58920
 keras/src/saving/serialization_lib.py,sha256=yzCTm8hin__MGA2N5M5F-8Zbts5ZJVmINbrH4wEtIwI,30334
-keras/src/testing/__init__.py,sha256=RQ5ZZ88NhcDTHCIpPulvaiTOTdJqAMH9ZhptXyMcqqY,368
-keras/src/testing/test_case.py,sha256=24aSbz5WG5ICUavaLWo-PWBD_O9T7feRGvrLoNS19Q8,31780
+keras/src/testing/__init__.py,sha256=29k8fvygLsF33eSTUowVIEEQLRB3AzTQRvYotEOe3U8,428
+keras/src/testing/test_case.py,sha256=FrQiXRvtarSv7G9g1PK8GJYJeMsKVg0GMzsLFivMopI,32432
 keras/src/testing/test_utils.py,sha256=6Vb8tJIyjU1ay63w3jvXNNhh7sSNrosQll4ii1NXELQ,6197
 keras/src/trainers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 keras/src/trainers/compile_utils.py,sha256=k5FDn7we0RN9fhRslY_WOQZRFwfzjqpYmiDeOKkAKqk,31260
@@ -618,7 +618,7 @@ keras/utils/bounding_boxes/__init__.py,sha256=jtvQll4u8ZY0Z96HwNhP1nxWEG9FM3gI-6
 keras/utils/legacy/__init__.py,sha256=oSYZz6uS8UxSElRaaJYWJEoweJ4GAasZjnn7fNaOlog,342
 keras/visualization/__init__.py,sha256=UKWmiy6sps4SWlmQi9WX8_Z53cPpLlphz2zIeHdwJpQ,722
 keras/wrappers/__init__.py,sha256=QkS-O5K8qGS7C3sytF8MpmO6PasATpNVGF8qtb7Ojsw,407
-keras_nightly-3.14.0.dev2026020304.dist-info/METADATA,sha256=hy2sly87dVSoOiutBAyDJXlZX_b0UIWgCmSfphO4lwU,6339
-keras_nightly-3.14.0.dev2026020304.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-keras_nightly-3.14.0.dev2026020304.dist-info/top_level.txt,sha256=ptcw_-QuGZ4ZDjMdwi_Z0clZm8QAqFdvzzFnDEOTs9o,6
-keras_nightly-3.14.0.dev2026020304.dist-info/RECORD,,
+keras_nightly-3.14.0.dev2026020404.dist-info/METADATA,sha256=0_dkofG5lLCYTwNn1VT_Inp2boYDQMIjPvIW5QwXM9I,6339
+keras_nightly-3.14.0.dev2026020404.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+keras_nightly-3.14.0.dev2026020404.dist-info/top_level.txt,sha256=ptcw_-QuGZ4ZDjMdwi_Z0clZm8QAqFdvzzFnDEOTs9o,6
+keras_nightly-3.14.0.dev2026020404.dist-info/RECORD,,