Publishing 2019 R3 content

[platform/upstream/dldt.git] / inference-engine / tests / unit / shape_infer / built_in_shape_infer_general_test.cpp

diff --git a/inference-engine/tests/unit/shape_infer/built_in_shape_infer_general_test.cpp b/inference-engine/tests/unit/shape_infer/built_in_shape_infer_general_test.cpp
index 6e1f7b5..cb9545e 100644 (file)
--- a/inference-engine/tests/unit/shape_infer/built_in_shape_infer_general_test.cpp
+++ b/inference-engine/tests/unit/shape_infer/built_in_shape_infer_general_test.cpp
@@ -790,6 +790,42 @@ INSTANTIATE_TEST_CASE_P(
                                                       {{2, 128, 10, 10}}}),
                                       MapParams(MapStrStr(std::map<std::string, std::string>{ {"levels", "2"}})),
                                       LayerDataName("data"),
+                                      CanInfer(true)),
+                ::testing::make_tuple(LayerType("Unique"),
+                                      InOutShapes({{{5}},
+                                                   {{5}, {5}}}),
+                                      NewInOutShapes({{{25}},
+                                                      {{25}, {25}}}),
+                                      MapParams(MapStrStr(std::map<std::string, std::string>{{"sorted", "false"},
+                                                                                             {"return_inverse", "true"},
+                                                                                             {"return_counts", "false"}})),
+                                      LayerDataName("data"),
+                                      CanInfer(true)),
+                ::testing::make_tuple(LayerType("Unique"),
+                                      InOutShapes({{{5}},
+                                                   {{5}, {5}, {5}}}),
+                                      NewInOutShapes({{{25}},
+                                                      {{25}, {25}, {25}}}),
+                                      MapParams(MapStrStr(std::map<std::string, std::string>{{"sorted", "false"},
+                                                                                             {"return_inverse", "true"},
+                                                                                             {"return_counts", "true"}})),
+                                      LayerDataName("data"),
+                                      CanInfer(true)),
+                ::testing::make_tuple(LayerType("Scatter"),
+                                      InOutShapes({{{3, 3}, {2, 3}},
+                                                   {{3,3}}}),
+                                      NewInOutShapes({{{4,  4}, {3, 4}},
+                                                      {{4,4}}}),
+                                      MapParams(MapStrStr(std::map<std::string, std::string>{{"axis", "0"}})),
+                                      LayerDataName("data"),
+                                      CanInfer(true)),
+                ::testing::make_tuple(LayerType("NonMaxSuppression"),
+                                      InOutShapes({{{1, 2, 4}, {1, 3, 2}},
+                                                   {{6, 3}}}),
+                                      NewInOutShapes({{{2, 5, 4}, {2, 3, 5}},
+                                                      {{30, 3}}}),
+                                      MapParams(MapStrStr(std::map<std::string, std::string>{{"center_point_box", "0"}})),
+                                      LayerDataName("data"),
                                       CanInfer(true))
         )
 );