OpenI
/
MegEngine
Not watched
1
0
Fork 0
Code
Releases 31 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Tree: c53cad2049
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'c53cad2049'
${ noResults }
MegEngine/dnn/test/common/padding.h

462 lines
18 kB
Raw Blame History 

  1. /**

  2.  * \file dnn/test/common/padding.h

  3.  * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")

  4.  *

  5.  * Copyright (c) 2014-2021 Megvii Inc. All rights reserved.

  6.  *

  7.  * Unless required by applicable law or agreed to in writing,

  8.  * software distributed under the License is distributed on an

  9.  * "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or

  10.  * implied.

  11.  */

  12. #pragma once

  13. #include <cstddef>

  14. #include <iostream>

  15. #include "megdnn/basic_types.h"

  16. #include "megdnn/opr_param_defs.h"

  17. 

  18. namespace megdnn {

  19. namespace test {

  20. namespace padding {

  21. 

  22. struct TestArg {

  23.     param::Padding param;

  24.     TensorShape src;

  25.     TensorShape dst;

  26.     TestArg(param::Padding _param, TensorShape _src, TensorShape _dst)

  27.             : param(_param), src(_src), dst(_dst) {}

  28. };

  29. 

  30. inline std::vector<TestArg> get_args() {

  31.     size_t src_shape_dim0 = 5;

  32.     size_t src_shape_dim1 = 5;

  33.     size_t src_shape_dim2 = 5;

  34.     size_t src_shape_dim3 = 5;

  35.     size_t src_shape_dim4 = 5;

  36.     size_t src_shape_dim5 = 5;

  37.     size_t src_shape_dim6 = 5;

  38. 

  39.     size_t dst_shape_dim0 = 8;

  40.     size_t dst_shape_dim1 = 8;

  41.     size_t dst_shape_dim2 = 8;

  42.     size_t dst_shape_dim3 = 8;

  43.     size_t dst_shape_dim4 = 8;

  44.     size_t dst_shape_dim5 = 8;

  45.     size_t dst_shape_dim6 = 8;

  46. 

  47.     std::vector<TestArg> args;

  48. 

  49.     param::Padding cur_param;

  50. 

  51.     cur_param.front_offset_dim0 = 0;

  52.     cur_param.front_offset_dim1 = 0;

  53.     cur_param.front_offset_dim2 = 0;

  54.     cur_param.front_offset_dim3 = 0;

  55.     cur_param.front_offset_dim4 = 0;

  56.     cur_param.front_offset_dim5 = 0;

  57.     cur_param.front_offset_dim6 = 0;

  58.     cur_param.back_offset_dim0 = 0;

  59.     cur_param.back_offset_dim1 = 0;

  60.     cur_param.back_offset_dim2 = 0;

  61.     cur_param.back_offset_dim3 = 0;

  62.     cur_param.back_offset_dim4 = 0;

  63.     cur_param.back_offset_dim5 = 0;

  64.     cur_param.back_offset_dim6 = 0;

  65. 

  66.     cur_param.padding_val = 2;

  67. 

  68.     cur_param.front_offset_dim0 = 1;

  69.     cur_param.back_offset_dim0 = 2;

  70. 

  71.     cur_param.padding_mode = param::Padding::PaddingMode::CONSTANT;

  72.     args.emplace_back(

  73.             cur_param, TensorShape{src_shape_dim0}, TensorShape{dst_shape_dim0});

  74. 

  75.     cur_param.padding_mode = param::Padding::PaddingMode::REPLICATE;

  76.     args.emplace_back(

  77.             cur_param, TensorShape{src_shape_dim0}, TensorShape{dst_shape_dim0});

  78. 

  79.     cur_param.padding_mode = param::Padding::PaddingMode::REFLECT;

  80.     args.emplace_back(

  81.             cur_param, TensorShape{src_shape_dim0}, TensorShape{dst_shape_dim0});

  82. 

  83.     cur_param.front_offset_dim1 = 2;

  84.     cur_param.back_offset_dim1 = 1;

  85. 

  86.     cur_param.padding_mode = param::Padding::PaddingMode::CONSTANT;

  87.     args.emplace_back(

  88.             cur_param, TensorShape{src_shape_dim0, src_shape_dim1},

  89.             TensorShape{dst_shape_dim0, dst_shape_dim1});

  90. 

  91.     cur_param.padding_mode = param::Padding::PaddingMode::REPLICATE;

  92.     args.emplace_back(

  93.             cur_param, TensorShape{src_shape_dim0, src_shape_dim1},

  94.             TensorShape{dst_shape_dim0, dst_shape_dim1});

  95. 

  96.     cur_param.padding_mode = param::Padding::PaddingMode::REFLECT;

  97.     args.emplace_back(

  98.             cur_param, TensorShape{src_shape_dim0, src_shape_dim1},

  99.             TensorShape{dst_shape_dim0, dst_shape_dim1});

  100. 

  101.     cur_param.front_offset_dim2 = 1;

  102.     cur_param.back_offset_dim2 = 2;

  103. 

  104.     cur_param.padding_mode = param::Padding::PaddingMode::CONSTANT;

  105.     args.emplace_back(

  106.             cur_param, TensorShape{src_shape_dim0, src_shape_dim1, src_shape_dim2},

  107.             TensorShape{dst_shape_dim0, dst_shape_dim1, dst_shape_dim2});

  108. 

  109.     cur_param.padding_mode = param::Padding::PaddingMode::REPLICATE;

  110.     args.emplace_back(

  111.             cur_param, TensorShape{src_shape_dim0, src_shape_dim1, src_shape_dim2},

  112.             TensorShape{dst_shape_dim0, dst_shape_dim1, dst_shape_dim2});

  113. 

  114.     cur_param.padding_mode = param::Padding::PaddingMode::REFLECT;

  115.     args.emplace_back(

  116.             cur_param, TensorShape{src_shape_dim0, src_shape_dim1, src_shape_dim2},

  117.             TensorShape{dst_shape_dim0, dst_shape_dim1, dst_shape_dim2});

  118. 

  119.     cur_param.front_offset_dim3 = 0;

  120.     cur_param.back_offset_dim3 = 3;

  121. 

  122.     cur_param.padding_mode = param::Padding::PaddingMode::CONSTANT;

  123.     args.emplace_back(

  124.             cur_param,

  125.             TensorShape{src_shape_dim0, src_shape_dim1, src_shape_dim2, src_shape_dim3},

  126.             TensorShape{

  127.                     dst_shape_dim0, dst_shape_dim1, dst_shape_dim2, dst_shape_dim3});

  128. 

  129.     cur_param.padding_mode = param::Padding::PaddingMode::REPLICATE;

  130.     args.emplace_back(

  131.             cur_param,

  132.             TensorShape{src_shape_dim0, src_shape_dim1, src_shape_dim2, src_shape_dim3},

  133.             TensorShape{

  134.                     dst_shape_dim0, dst_shape_dim1, dst_shape_dim2, dst_shape_dim3});

  135. 

  136.     cur_param.padding_mode = param::Padding::PaddingMode::REFLECT;

  137.     args.emplace_back(

  138.             cur_param,

  139.             TensorShape{src_shape_dim0, src_shape_dim1, src_shape_dim2, src_shape_dim3},

  140.             TensorShape{

  141.                     dst_shape_dim0, dst_shape_dim1, dst_shape_dim2, dst_shape_dim3});

  142. 

  143.     cur_param.front_offset_dim4 = 3;

  144.     cur_param.back_offset_dim4 = 0;

  145. 

  146.     cur_param.padding_mode = param::Padding::PaddingMode::CONSTANT;

  147.     args.emplace_back(

  148.             cur_param,

  149.             TensorShape{

  150.                     src_shape_dim0, src_shape_dim1, src_shape_dim2, src_shape_dim3,

  151.                     src_shape_dim4},

  152.             TensorShape{

  153.                     dst_shape_dim0, dst_shape_dim1, dst_shape_dim2, dst_shape_dim3,

  154.                     dst_shape_dim4});

  155. 

  156.     cur_param.padding_mode = param::Padding::PaddingMode::REPLICATE;

  157.     args.emplace_back(

  158.             cur_param,

  159.             TensorShape{

  160.                     src_shape_dim0, src_shape_dim1, src_shape_dim2, src_shape_dim3,

  161.                     src_shape_dim4},

  162.             TensorShape{

  163.                     dst_shape_dim0, dst_shape_dim1, dst_shape_dim2, dst_shape_dim3,

  164.                     dst_shape_dim4});

  165. 

  166.     cur_param.padding_mode = param::Padding::PaddingMode::REFLECT;

  167.     args.emplace_back(

  168.             cur_param,

  169.             TensorShape{

  170.                     src_shape_dim0, src_shape_dim1, src_shape_dim2, src_shape_dim3,

  171.                     src_shape_dim4},

  172.             TensorShape{

  173.                     dst_shape_dim0, dst_shape_dim1, dst_shape_dim2, dst_shape_dim3,

  174.                     dst_shape_dim4});

  175. 

  176.     cur_param.front_offset_dim5 = 1;

  177.     cur_param.back_offset_dim5 = 2;

  178. 

  179.     cur_param.padding_mode = param::Padding::PaddingMode::CONSTANT;

  180.     args.emplace_back(

  181.             cur_param,

  182.             TensorShape{

  183.                     src_shape_dim0, src_shape_dim1, src_shape_dim2, src_shape_dim3,

  184.                     src_shape_dim4, src_shape_dim5},

  185.             TensorShape{

  186.                     dst_shape_dim0, dst_shape_dim1, dst_shape_dim2, dst_shape_dim3,

  187.                     dst_shape_dim4, dst_shape_dim5});

  188. 

  189.     cur_param.padding_mode = param::Padding::PaddingMode::REPLICATE;

  190.     args.emplace_back(

  191.             cur_param,

  192.             TensorShape{

  193.                     src_shape_dim0, src_shape_dim1, src_shape_dim2, src_shape_dim3,

  194.                     src_shape_dim4, src_shape_dim5},

  195.             TensorShape{

  196.                     dst_shape_dim0, dst_shape_dim1, dst_shape_dim2, dst_shape_dim3,

  197.                     dst_shape_dim4, dst_shape_dim5});

  198. 

  199.     cur_param.padding_mode = param::Padding::PaddingMode::REFLECT;

  200.     args.emplace_back(

  201.             cur_param,

  202.             TensorShape{

  203.                     src_shape_dim0, src_shape_dim1, src_shape_dim2, src_shape_dim3,

  204.                     src_shape_dim4, src_shape_dim5},

  205.             TensorShape{

  206.                     dst_shape_dim0, dst_shape_dim1, dst_shape_dim2, dst_shape_dim3,

  207.                     dst_shape_dim4, dst_shape_dim5});

  208. 

  209.     cur_param.front_offset_dim6 = 0;

  210.     cur_param.front_offset_dim6 = 3;

  211. 

  212.     cur_param.padding_mode = param::Padding::PaddingMode::CONSTANT;

  213.     args.emplace_back(

  214.             cur_param,

  215.             TensorShape{

  216.                     src_shape_dim0, src_shape_dim1, src_shape_dim2, src_shape_dim3,

  217.                     src_shape_dim4, src_shape_dim5, src_shape_dim6},

  218.             TensorShape{

  219.                     dst_shape_dim0, dst_shape_dim1, dst_shape_dim2, dst_shape_dim3,

  220.                     dst_shape_dim4, dst_shape_dim5, dst_shape_dim6});

  221. 

  222.     cur_param.padding_mode = param::Padding::PaddingMode::REPLICATE;

  223.     args.emplace_back(

  224.             cur_param,

  225.             TensorShape{

  226.                     src_shape_dim0, src_shape_dim1, src_shape_dim2, src_shape_dim3,

  227.                     src_shape_dim4, src_shape_dim5, src_shape_dim6},

  228.             TensorShape{

  229.                     dst_shape_dim0, dst_shape_dim1, dst_shape_dim2, dst_shape_dim3,

  230.                     dst_shape_dim4, dst_shape_dim5, dst_shape_dim6});

  231. 

  232.     cur_param.padding_mode = param::Padding::PaddingMode::REFLECT;

  233.     args.emplace_back(

  234.             cur_param,

  235.             TensorShape{

  236.                     src_shape_dim0, src_shape_dim1, src_shape_dim2, src_shape_dim3,

  237.                     src_shape_dim4, src_shape_dim5, src_shape_dim6},

  238.             TensorShape{

  239.                     dst_shape_dim0, dst_shape_dim1, dst_shape_dim2, dst_shape_dim3,

  240.                     dst_shape_dim4, dst_shape_dim5, dst_shape_dim6});

  241. 

  242.     return args;

  243. }

  244. 

  245. inline std::vector<TestArg> get_args_backward() {

  246.     size_t src_shape_dim0 = 8;

  247.     size_t src_shape_dim1 = 8;

  248.     size_t src_shape_dim2 = 8;

  249.     size_t src_shape_dim3 = 8;

  250.     size_t src_shape_dim4 = 8;

  251.     size_t src_shape_dim5 = 8;

  252.     size_t src_shape_dim6 = 8;

  253. 

  254.     size_t dst_shape_dim0 = 5;

  255.     size_t dst_shape_dim1 = 5;

  256.     size_t dst_shape_dim2 = 5;

  257.     size_t dst_shape_dim3 = 5;

  258.     size_t dst_shape_dim4 = 5;

  259.     size_t dst_shape_dim5 = 5;

  260.     size_t dst_shape_dim6 = 5;

  261. 

  262.     std::vector<TestArg> args;

  263. 

  264.     param::Padding cur_param;

  265. 

  266.     cur_param.front_offset_dim0 = 0;

  267.     cur_param.front_offset_dim1 = 0;

  268.     cur_param.front_offset_dim2 = 0;

  269.     cur_param.front_offset_dim3 = 0;

  270.     cur_param.front_offset_dim4 = 0;

  271.     cur_param.front_offset_dim5 = 0;

  272.     cur_param.front_offset_dim6 = 0;

  273.     cur_param.back_offset_dim0 = 0;

  274.     cur_param.back_offset_dim1 = 0;

  275.     cur_param.back_offset_dim2 = 0;

  276.     cur_param.back_offset_dim3 = 0;

  277.     cur_param.back_offset_dim4 = 0;

  278.     cur_param.back_offset_dim5 = 0;

  279.     cur_param.back_offset_dim6 = 0;

  280. 

  281.     cur_param.padding_val = 2;

  282. 

  283.     cur_param.front_offset_dim0 = 1;

  284.     cur_param.back_offset_dim0 = 2;

  285. 

  286.     cur_param.padding_mode = param::Padding::PaddingMode::CONSTANT;

  287.     args.emplace_back(

  288.             cur_param, TensorShape{src_shape_dim0}, TensorShape{dst_shape_dim0});

  289. 

  290.     cur_param.padding_mode = param::Padding::PaddingMode::REPLICATE;

  291.     args.emplace_back(

  292.             cur_param, TensorShape{src_shape_dim0}, TensorShape{dst_shape_dim0});

  293. 

  294.     cur_param.padding_mode = param::Padding::PaddingMode::REFLECT;

  295.     args.emplace_back(

  296.             cur_param, TensorShape{src_shape_dim0}, TensorShape{dst_shape_dim0});

  297. 

  298.     cur_param.front_offset_dim1 = 2;

  299.     cur_param.back_offset_dim1 = 1;

  300. 

  301.     cur_param.padding_mode = param::Padding::PaddingMode::CONSTANT;

  302.     args.emplace_back(

  303.             cur_param, TensorShape{src_shape_dim0, src_shape_dim1},

  304.             TensorShape{dst_shape_dim0, dst_shape_dim1});

  305. 

  306.     cur_param.padding_mode = param::Padding::PaddingMode::REPLICATE;

  307.     args.emplace_back(

  308.             cur_param, TensorShape{src_shape_dim0, src_shape_dim1},

  309.             TensorShape{dst_shape_dim0, dst_shape_dim1});

  310. 

  311.     cur_param.padding_mode = param::Padding::PaddingMode::REFLECT;

  312.     args.emplace_back(

  313.             cur_param, TensorShape{src_shape_dim0, src_shape_dim1},

  314.             TensorShape{dst_shape_dim0, dst_shape_dim1});

  315. 

  316.     cur_param.front_offset_dim2 = 1;

  317.     cur_param.back_offset_dim2 = 2;

  318. 

  319.     cur_param.padding_mode = param::Padding::PaddingMode::CONSTANT;

  320.     args.emplace_back(

  321.             cur_param, TensorShape{src_shape_dim0, src_shape_dim1, src_shape_dim2},

  322.             TensorShape{dst_shape_dim0, dst_shape_dim1, dst_shape_dim2});

  323. 

  324.     cur_param.padding_mode = param::Padding::PaddingMode::REPLICATE;

  325.     args.emplace_back(

  326.             cur_param, TensorShape{src_shape_dim0, src_shape_dim1, src_shape_dim2},

  327.             TensorShape{dst_shape_dim0, dst_shape_dim1, dst_shape_dim2});

  328. 

  329.     cur_param.padding_mode = param::Padding::PaddingMode::REFLECT;

  330.     args.emplace_back(

  331.             cur_param, TensorShape{src_shape_dim0, src_shape_dim1, src_shape_dim2},

  332.             TensorShape{dst_shape_dim0, dst_shape_dim1, dst_shape_dim2});

  333. 

  334.     cur_param.front_offset_dim3 = 0;

  335.     cur_param.back_offset_dim3 = 3;

  336. 

  337.     cur_param.padding_mode = param::Padding::PaddingMode::CONSTANT;

  338.     args.emplace_back(

  339.             cur_param,

  340.             TensorShape{src_shape_dim0, src_shape_dim1, src_shape_dim2, src_shape_dim3},

  341.             TensorShape{

  342.                     dst_shape_dim0, dst_shape_dim1, dst_shape_dim2, dst_shape_dim3});

  343. 

  344.     cur_param.padding_mode = param::Padding::PaddingMode::REPLICATE;

  345.     args.emplace_back(

  346.             cur_param,

  347.             TensorShape{src_shape_dim0, src_shape_dim1, src_shape_dim2, src_shape_dim3},

  348.             TensorShape{

  349.                     dst_shape_dim0, dst_shape_dim1, dst_shape_dim2, dst_shape_dim3});

  350. 

  351.     cur_param.padding_mode = param::Padding::PaddingMode::REFLECT;

  352.     args.emplace_back(

  353.             cur_param,

  354.             TensorShape{src_shape_dim0, src_shape_dim1, src_shape_dim2, src_shape_dim3},

  355.             TensorShape{

  356.                     dst_shape_dim0, dst_shape_dim1, dst_shape_dim2, dst_shape_dim3});

  357. 

  358.     cur_param.front_offset_dim4 = 3;

  359.     cur_param.back_offset_dim4 = 0;

  360. 

  361.     cur_param.padding_mode = param::Padding::PaddingMode::CONSTANT;

  362.     args.emplace_back(

  363.             cur_param,

  364.             TensorShape{

  365.                     src_shape_dim0, src_shape_dim1, src_shape_dim2, src_shape_dim3,

  366.                     src_shape_dim4},

  367.             TensorShape{

  368.                     dst_shape_dim0, dst_shape_dim1, dst_shape_dim2, dst_shape_dim3,

  369.                     dst_shape_dim4});

  370. 

  371.     cur_param.padding_mode = param::Padding::PaddingMode::REPLICATE;

  372.     args.emplace_back(

  373.             cur_param,

  374.             TensorShape{

  375.                     src_shape_dim0, src_shape_dim1, src_shape_dim2, src_shape_dim3,

  376.                     src_shape_dim4},

  377.             TensorShape{

  378.                     dst_shape_dim0, dst_shape_dim1, dst_shape_dim2, dst_shape_dim3,

  379.                     dst_shape_dim4});

  380. 

  381.     cur_param.padding_mode = param::Padding::PaddingMode::REFLECT;

  382.     args.emplace_back(

  383.             cur_param,

  384.             TensorShape{

  385.                     src_shape_dim0, src_shape_dim1, src_shape_dim2, src_shape_dim3,

  386.                     src_shape_dim4},

  387.             TensorShape{

  388.                     dst_shape_dim0, dst_shape_dim1, dst_shape_dim2, dst_shape_dim3,

  389.                     dst_shape_dim4});

  390. 

  391.     cur_param.front_offset_dim5 = 1;

  392.     cur_param.back_offset_dim5 = 2;

  393. 

  394.     cur_param.padding_mode = param::Padding::PaddingMode::CONSTANT;

  395.     args.emplace_back(

  396.             cur_param,

  397.             TensorShape{

  398.                     src_shape_dim0, src_shape_dim1, src_shape_dim2, src_shape_dim3,

  399.                     src_shape_dim4, src_shape_dim5},

  400.             TensorShape{

  401.                     dst_shape_dim0, dst_shape_dim1, dst_shape_dim2, dst_shape_dim3,

  402.                     dst_shape_dim4, dst_shape_dim5});

  403. 

  404.     cur_param.padding_mode = param::Padding::PaddingMode::REPLICATE;

  405.     args.emplace_back(

  406.             cur_param,

  407.             TensorShape{

  408.                     src_shape_dim0, src_shape_dim1, src_shape_dim2, src_shape_dim3,

  409.                     src_shape_dim4, src_shape_dim5},

  410.             TensorShape{

  411.                     dst_shape_dim0, dst_shape_dim1, dst_shape_dim2, dst_shape_dim3,

  412.                     dst_shape_dim4, dst_shape_dim5});

  413. 

  414.     cur_param.padding_mode = param::Padding::PaddingMode::REFLECT;

  415.     args.emplace_back(

  416.             cur_param,

  417.             TensorShape{

  418.                     src_shape_dim0, src_shape_dim1, src_shape_dim2, src_shape_dim3,

  419.                     src_shape_dim4, src_shape_dim5},

  420.             TensorShape{

  421.                     dst_shape_dim0, dst_shape_dim1, dst_shape_dim2, dst_shape_dim3,

  422.                     dst_shape_dim4, dst_shape_dim5});

  423. 

  424.     cur_param.front_offset_dim6 = 0;

  425.     cur_param.back_offset_dim6 = 3;

  426. 

  427.     cur_param.padding_mode = param::Padding::PaddingMode::CONSTANT;

  428.     args.emplace_back(

  429.             cur_param,

  430.             TensorShape{

  431.                     src_shape_dim0, src_shape_dim1, src_shape_dim2, src_shape_dim3,

  432.                     src_shape_dim4, src_shape_dim5, src_shape_dim6},

  433.             TensorShape{

  434.                     dst_shape_dim0, dst_shape_dim1, dst_shape_dim2, dst_shape_dim3,

  435.                     dst_shape_dim4, dst_shape_dim5, dst_shape_dim6});

  436. 

  437.     cur_param.padding_mode = param::Padding::PaddingMode::REPLICATE;

  438.     args.emplace_back(

  439.             cur_param,

  440.             TensorShape{

  441.                     src_shape_dim0, src_shape_dim1, src_shape_dim2, src_shape_dim3,

  442.                     src_shape_dim4, src_shape_dim5, src_shape_dim6},

  443.             TensorShape{

  444.                     dst_shape_dim0, dst_shape_dim1, dst_shape_dim2, dst_shape_dim3,

  445.                     dst_shape_dim4, dst_shape_dim5, dst_shape_dim6});

  446. 

  447.     cur_param.padding_mode = param::Padding::PaddingMode::REFLECT;

  448.     args.emplace_back(

  449.             cur_param,

  450.             TensorShape{

  451.                     src_shape_dim0, src_shape_dim1, src_shape_dim2, src_shape_dim3,

  452.                     src_shape_dim4, src_shape_dim5, src_shape_dim6},

  453.             TensorShape{

  454.                     dst_shape_dim0, dst_shape_dim1, dst_shape_dim2, dst_shape_dim3,

  455.                     dst_shape_dim4, dst_shape_dim5, dst_shape_dim6});

  456. 

  457.     return args;

  458. }

  459. 

  460. }  // namespace padding

  461. }  // namespace test

  462. }  // namespace megdnn