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

Enable GatherElements tests on IE_CPU (#4044)

Browse Source
This commit is contained in:
Tomasz Jankowski 2021-02-02 12:59:44 +01:00 committed by GitHub
parent 38fab0265d
commit cc000e57e0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 13 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
ngraph/test/backend/gather_elements.in.cpp
View File

@ -368,15 +368,15 @@ NGRAPH_TEST(${BACKEND_NAME}, evaluate_2D_gather_elements_2x2x1_data_float32)
test_case.add_input<int32_t>(data);
test_case.add_input<int32_t>(indices);
test_case.add_expected_output<float>(vector<float>{5,
4,
1,
4,
test_case.add_expected_output<int32_t>({5,
4,
1,
4,
1,
4,
5,
4});
1,
4,
5,
4});
// clang-format on
test_case.run();
}

21
ngraph/test/runtime/ie/unit_test.manifest
View File

@ -1559,25 +1559,14 @@ onnx_controlflow_loop_infinite
onnx_dyn_shapes_reduce_max_dynamic_input_rank_negative_axis
IE_GPU.range_v4_trunc_inputs
# not implemented yet on CPU and GPU plugins
IE_CPU.evaluate_1D_gather_elements_3_indices_int32
IE_CPU.evaluate_2D_gather_elements_2x2_indices_int32_axis_0
IE_CPU.evaluate_2D_gather_elements_2x2_indices_int32_axis_1
IE_CPU.evaluate_2D_gather_elements_2x2_indices_int32_axis_minus_1
IE_CPU.evaluate_2D_gather_elements_2x3_indices_int32
IE_CPU.evaluate_3D_gather_elements_3x2x2_indices_int32
IE_CPU.evaluate_3D_gather_elements_3x2x2_indices_int64
# CPU plugin does not support bool type
IE_CPU.evaluate_2D_gather_elements_3x2_data_bool
IE_CPU.evaluate_2D_gather_elements_2x3_data_float32
# CPU plugin does not validate whether indices values are out of range
IE_CPU.evaluate_1D_gather_elements_negative_test
IE_CPU.evaluate_2D_gather_elements_negative_test
IE_CPU.evaluate_2D_gather_elements_2x2x1_data_float32
IE_CPU.evaluate_4D_gather_elements_3x2x2x2_indices_int64
IE_CPU.onnx_model_gather_elements_float_1D
IE_CPU.onnx_model_gather_elements_float_negative_axis
IE_CPU.onnx_model_gather_elements_int32_axis_0
IE_CPU.onnx_model_gather_elements_int8_axis_1
IE_CPU.onnx_model_gather_elements_float_3D_axis_2
# not implemented yet on GPU plugin
IE_GPU.evaluate_1D_gather_elements_3_indices_int32
IE_GPU.evaluate_2D_gather_elements_2x2_indices_int32_axis_0
IE_GPU.evaluate_2D_gather_elements_2x2_indices_int32_axis_1