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

Fix broken op create test in python api

Browse Source
This commit is contained in:
ggalieroc
2021-03-12 18:43:00 +01:00
parent 944af98b8c
commit 72fbfc2967
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
ngraph/python/tests/test_ngraph/test_create_op.py
View File

@@ -117,7 +117,7 @@ def test_ctc_greedy_decoder_seq_len(fp_dtype, int_dtype, int_ci, int_sl, merge_r
assert list(node.get_output_shape(0)) == expected_shape
@pytest.mark.parametrize("dtype", np_types)
@pytest.mark.parametrize("dtype", [np.float32, np.float64])
def test_deformable_convolution(dtype):
strides = np.array([1, 1])
pads_begin = np.array([0, 0])
@@ -125,7 +125,7 @@ def test_deformable_convolution(dtype):
dilations = np.array([1, 1])
input0_shape = [1, 1, 9, 9]
input1_shape = [1, 1, 9, 9]
input1_shape = [1, 18, 9, 9]
input2_shape = [1, 1, 3, 3]
expected_shape = [1, 1, 7, 7]