Skip to content

Commit

Permalink
fixed test_all
Browse files Browse the repository at this point in the history
  • Loading branch information
mryzhov committed Oct 24, 2023
1 parent c8494da commit b7ecd4f
Showing 1 changed file with 12 additions and 12 deletions.
24 changes: 12 additions & 12 deletions tests/layer_tests/pytorch_tests/test_all.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,12 +37,12 @@ def _prepare_input(self):
return (self.input_tensor,)

@pytest.mark.parametrize("input_shape, d_type", [
(np.eye(5,5), np.int16),
(np.zeros((5, 5)), np.int16),
(np.zeros((9,8)) + 1, np.int16),
([5, 9, 7], np.int16),
([10, 13, 11], np.int16),
([8, 7, 6, 5, 4], np.int16),
(np.eye(5,5), np.int64),
(np.zeros((5, 5)), np.int64),
(np.zeros((9,8)) + 1, np.int64),
([5, 9, 7], np.int64),
([10, 13, 11], np.int64),
([8, 7, 6, 5, 4], np.int64),
([11, 11], np.uint8),
([7, 7], np.uint8)
])
Expand All @@ -57,12 +57,12 @@ def test_all_noparams(self, input_shape, d_type, ie_device, precision, ir_versio
ie_device, precision, ir_version, trace_model=True, freeze_model=False)

@pytest.mark.parametrize("input_shape, d_type", [
(np.eye(5,5), np.int16),
(np.zeros((5, 5)), np.int16),
(np.zeros((9,8)) + 1, np.int16),
([5, 9, 7], np.int16),
([10, 13, 11], np.int16),
([8, 7, 6, 5, 4], np.int16),
(np.eye(5,5), np.int64),
(np.zeros((5, 5)), np.int64),
(np.zeros((9,8)) + 1, np.int64),
([5, 9, 7], np.int64),
([10, 13, 11], np.int64),
([8, 7, 6, 5, 4], np.int64),
([11, 11], np.uint8),
([7, 7], np.uint8)
])
Expand Down

0 comments on commit b7ecd4f

Please sign in to comment.