Skip to content

Commit 11cd909

Browse files
committed
[OpenVINO Backend] numpy.empty implementation
1 parent 3820c0f commit 11cd909

File tree

2 files changed

+12
-4
lines changed

2 files changed

+12
-4
lines changed

keras/src/backend/openvino/excluded_concrete_tests.txt

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,6 @@ NumpyDtypeTest::test_diag
2222
NumpyDtypeTest::test_diff
2323
NumpyDtypeTest::test_digitize
2424
NumpyDtypeTest::test_einsum
25-
NumpyDtypeTest::test_empty
2625
NumpyDtypeTest::test_exp2
2726
NumpyDtypeTest::test_expm1
2827
NumpyDtypeTest::test_eye

keras/src/backend/openvino/numpy.py

Lines changed: 12 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -362,7 +362,7 @@ def average(x, axis=None, weights=None):
362362
x_type = x.get_element_type()
363363
weights_type = weights.get_element_type()
364364
if (weights_type.is_integral() or weights_type == Type.boolean) and (
365-
x_type.is_integral() or x_type == Type.boolean
365+
x_type.is_integral() or x_type == Type.boolean
366366
):
367367
x = ov_opset.convert(x, Type.f32).output(0)
368368
weights = ov_opset.convert(weights, Type.f32).output(0)
@@ -565,7 +565,16 @@ def dot(x, y):
565565

566566

567567
def empty(shape, dtype=None):
568-
raise NotImplementedError("`empty` is not supported with openvino backend")
568+
if dtype is not None:
569+
ov_type = OPENVINO_DTYPES[standardize_dtype(dtype)]
570+
else:
571+
ov_type = Type.f32
572+
if isinstance(shape, tuple):
573+
shape = list(shape)
574+
elif isinstance(shape, int):
575+
shape = [shape]
576+
empty_tensor = ov_opset.parameter(shape, ov_type).output(0)
577+
return OpenVINOKerasTensor(empty_tensor)
569578

570579

571580
def equal(x1, x2):
@@ -712,7 +721,7 @@ def less_equal(x1, x2):
712721

713722

714723
def linspace(
715-
start, stop, num=50, endpoint=True, retstep=False, dtype=None, axis=0
724+
start, stop, num=50, endpoint=True, retstep=False, dtype=None, axis=0
716725
):
717726
raise NotImplementedError(
718727
"`linspace` is not supported with openvino backend"

0 commit comments

Comments
 (0)