Skip to content
Snippets Groups Projects
Commit 3856a746 authored by Michal Szczepanski's avatar Michal Szczepanski
Browse files

section name modification - bug correction

parent d80e0404
No related branches found
No related tags found
1 merge request!155[Add] Resize Operator unit-tests
...@@ -38,7 +38,7 @@ TEST_CASE("[core/operator] Resize_Op(forwardDims)", "[Resize][forwardDimsScales] ...@@ -38,7 +38,7 @@ TEST_CASE("[core/operator] Resize_Op(forwardDims)", "[Resize][forwardDimsScales]
// std::shared_ptr<Tensor> input_roi = std::make_shared<Tensor>(); // std::shared_ptr<Tensor> input_roi = std::make_shared<Tensor>();
// op -> associateInput(1,input_roi); // op -> associateInput(1,input_roi);
SECTION("fix_sizes 1.") { SECTION("fix_scales upscale 1.") {
input_data->resize({1,1,2,2}); input_data->resize({1,1,2,2});
input_data->zeros(); input_data->zeros();
op -> associateInput(0,input_data); op -> associateInput(0,input_data);
...@@ -56,7 +56,7 @@ TEST_CASE("[core/operator] Resize_Op(forwardDims)", "[Resize][forwardDimsScales] ...@@ -56,7 +56,7 @@ TEST_CASE("[core/operator] Resize_Op(forwardDims)", "[Resize][forwardDimsScales]
fmt::print("{} = {}\n",op->getOutput(0)->dims()[i], dst_dimsOut->dims()[i]); fmt::print("{} = {}\n",op->getOutput(0)->dims()[i], dst_dimsOut->dims()[i]);
} }
} }
SECTION("fix_sizes upscale2.") { SECTION("fix_scales upscale 2.") {
input_data->resize({4,4,10,10}); input_data->resize({4,4,10,10});
input_data->zeros(); input_data->zeros();
op -> associateInput(0,input_data); op -> associateInput(0,input_data);
...@@ -74,7 +74,7 @@ TEST_CASE("[core/operator] Resize_Op(forwardDims)", "[Resize][forwardDimsScales] ...@@ -74,7 +74,7 @@ TEST_CASE("[core/operator] Resize_Op(forwardDims)", "[Resize][forwardDimsScales]
fmt::print("{} = {}\n",op->getOutput(0)->dims()[i], dst_dimsOut->dims()[i]); fmt::print("{} = {}\n",op->getOutput(0)->dims()[i], dst_dimsOut->dims()[i]);
} }
} }
SECTION("fix_sizes downscale 3.") { SECTION("fix_scales downscale 3.") {
input_data->resize({4,2,10,10}); input_data->resize({4,2,10,10});
input_data->zeros(); input_data->zeros();
op -> associateInput(0,input_data); op -> associateInput(0,input_data);
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment