| 84 | } |
| 85 | |
| 86 | std::vector<TestArg> get_args_1x1() { |
| 87 | std::vector<TestArg> args; |
| 88 | param::ConvBias cur_param; |
| 89 | using NLMode = param::ConvBias::NonlineMode; |
| 90 | |
| 91 | for (auto nlmode : |
| 92 | {NLMode::IDENTITY, NLMode::RELU, NLMode::SIGMOID, NLMode::H_SWISH}) { |
| 93 | cur_param.nonlineMode = nlmode; |
| 94 | for (size_t i : {16, 19}) { |
| 95 | cur_param.mode = param::ConvBias::Mode::CONVOLUTION; |
| 96 | args.emplace_back( |
| 97 | cur_param, TensorShape{2, 20, i, i + 1}, TensorShape{30, 20, 1, 1}, |
| 98 | TensorShape{1, 30, 1, 1}); |
| 99 | |
| 100 | cur_param.mode = param::ConvBias::Mode::CROSS_CORRELATION; |
| 101 | args.emplace_back( |
| 102 | cur_param, TensorShape{2, 20, i, i + 1}, TensorShape{30, 20, 1, 1}, |
| 103 | TensorShape{1, 30, 1, 1}); |
| 104 | } |
| 105 | } |
| 106 | return args; |
| 107 | } |
| 108 | |
| 109 | std::vector<TestArg> get_winograd_args(size_t kernel_size) { |
| 110 | std::vector<TestArg> args; |