diff options
Diffstat (limited to 'inference-engine/tests/unit/shape_infer/built_in_shape_infer_general_test.cpp')
-rw-r--r-- | inference-engine/tests/unit/shape_infer/built_in_shape_infer_general_test.cpp | 118 |
1 files changed, 110 insertions, 8 deletions
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 9508b952e..c7fc06eb5 100644 --- 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 @@ -192,6 +192,38 @@ INSTANTIATE_TEST_CASE_P( {"aspect_ratio", "2"}})), LayerDataName("data"), CanInfer(true)), + ::testing::make_tuple(LayerType("PriorBox"), + InOutShapes({{{2, 512, 32, 32}, {2, 3, 512, 512}}, + {{1, 2, 16384}}}), + NewInOutShapes({{{2, 512, 32, 32}, {2, 3, 512, 512}}, + {{1, 2, 16384}}}), + MapParams(MapStrStr( + std::map<std::string, std::string>{{"min_size", "35.84,52.46464"}, + {"max_size", ""}, + {"clip", "0"}, + {"step", "16"}, + {"flip", "0"}, + {"offset", "0.5"}, + {"aspect_ratio", "1.0,2.0,0.5"}, + {"scale_all_sizes", "0"}})), + LayerDataName("data"), + CanInfer(true)), + ::testing::make_tuple(LayerType("PriorBox"), + InOutShapes({{{2, 512, 32, 32}, {2, 3, 512, 512}}, + {{1, 2, 32768}}}), + NewInOutShapes({{{2, 512, 32, 32}, {2, 3, 512, 512}}, + {{1, 2, 28672}}}), + MapParams(MapStrStr( + std::map<std::string, std::string>{{"min_size", "35.84,52.46464"}, + {"max_size", ""}, + {"clip", "0"}, + {"step", "16"}, + {"offset", "0.5"}, + {"flip", "1"}, + {"aspect_ratio", "1.0,2.0,0.5"}, + {"scale_all_sizes", "0"}})), + LayerDataName("data"), + CanInfer(true)), ::testing::make_tuple(LayerType("DetectionOutput"), InOutShapes({{{2, 1, 4, 5}}, {{2, 1, 200, 7}}}), @@ -202,6 +234,53 @@ INSTANTIATE_TEST_CASE_P( {"nms_threshold", "0.44999998807907104"}})), LayerDataName("data"), CanInfer(true)), + ::testing::make_tuple(LayerType("Interp"), + InOutShapes({{{2, 2, 33, 65}}, + {{2, 2, 257, 513}}}), + NewInOutShapes({{{2, 2, 33, 65}}, + {{2, 2, 257, 513}}}), + MapParams(MapStrStr(std::map<std::string, std::string>{{"align_corners", "1"}, + {"height", "257"}, + {"pad_beg", "0"}, + {"pad_end", "0"}, + {"width", "513"}})), + LayerDataName("data"), + CanInfer(true)), + ::testing::make_tuple(LayerType("Interp"), + InOutShapes({{{2, 2, 33, 65}}, + {{2, 2, 66, 513}}}), + NewInOutShapes({{{2, 2, 33, 65}}, + {{2, 2, 66, 513}}}), + MapParams(MapStrStr(std::map<std::string, std::string>{{"align_corners", "1"}, + {"factor", "2"}, + {"width", "513"}, + {"pad_beg", "0"}, + {"pad_end", "0"}})), + LayerDataName("data"), + CanInfer(true)), + ::testing::make_tuple(LayerType("Interp"), + InOutShapes({{{2, 2, 33, 65}}, + {{2, 2, 257, 130}}}), + NewInOutShapes({{{2, 2, 33, 65}}, + {{2, 2, 257, 130}}}), + MapParams(MapStrStr(std::map<std::string, std::string>{{"align_corners", "1"}, + {"factor", "2"}, + {"height", "257"}, + {"pad_beg", "0"}, + {"pad_end", "0"}})), + LayerDataName("data"), + CanInfer(true)), + ::testing::make_tuple(LayerType("Interp"), + InOutShapes({{{2, 2, 33, 65}}, + {{2, 2, 257, 130}}}), + NewInOutShapes({{{2, 2, 33, 65}}, + {{2, 2, 257, 130}}}), + MapParams(MapStrStr(std::map<std::string, std::string>{{"align_corners", "1"}, + {"width", "513"}, + {"pad_beg", "0"}, + {"pad_end", "0"}})), + LayerDataName("data"), + CanInfer(false)), ::testing::make_tuple(LayerType("ROIPooling"), InOutShapes({{{2, 3, 4, 5}, {150, 5}}, {{150, 3, 6, 6}}}), @@ -256,6 +335,37 @@ INSTANTIATE_TEST_CASE_P( {"offset", "2,1"}})), LayerDataName("data"), CanInfer(true)), + ::testing::make_tuple(LayerType("Crop"), + InOutShapes({{{1, 3, 5, 5}}, + {{1, 5, 2, 3}}}), + NewInOutShapes({{{2, 1, 5, 6}}, + {{2, 1, 2, 1}}}), + MapParams(MapStrStr(std::map<std::string, std::string>{{"axis", "2,3"}, + {"offset", "2,1"}, + {"dim", "2,1"}})), + LayerDataName("data"), + CanInfer(true)), + // offset is too big + ::testing::make_tuple(LayerType("Crop"), + InOutShapes({{{1, 3, 5, 5}}, + {{1, 5, 2, 3}}}), + NewInOutShapes({{{2, 1, 5, 4}}, + {{2, 1, 2, 1}}}), + MapParams(MapStrStr(std::map<std::string, std::string>{{"axis", "2,3"}, + {"offset", "3,4"}, + {"dim", "2,1"}})), + LayerDataName("data"), + CanInfer(false)), + ::testing::make_tuple(LayerType("Crop"), + InOutShapes({{{1, 3, 5, 5}}, + {{1, 5, 2, 3}}}), + NewInOutShapes({{{2, 1, 5, 6}}, + {{2, 1, 1, 4}}}), + MapParams(MapStrStr(std::map<std::string, std::string>{{"axis", "2,3"}, + {"crop_begin", "2,1"}, + {"crop_end", "2,1"}})), + LayerDataName("data"), + CanInfer(true)), ::testing::make_tuple(LayerType("Tile"), InOutShapes({{{1, 3, 5, 5}}, {{1, 9, 5, 5}}}), @@ -283,14 +393,6 @@ INSTANTIATE_TEST_CASE_P( {"out_sizes", "2,4"}})), LayerDataName("data"), CanInfer(true)), - ::testing::make_tuple(LayerType("SpatialTransformer"), - InOutShapes({{{1, 6, 5, 5}, {1, 3}}, - {{1, 6, 5, 5}}}), - NewInOutShapes({{{2, 6, 5, 6}, {1, 3}}, - {{2, 6, 5, 6}}}), - MapParams(MapStrStr()), - LayerDataName("data"), - CanInfer(true)), ::testing::make_tuple(LayerType("CTCGreedyDecoder"), InOutShapes({{{88, 1, 48, 1}}, {{1, 88, 1, 1}}}), |