| 352 | } |
| 353 | |
| 354 | std::vector<TestArg> get_quantized_args() { |
| 355 | using NLMode = param::ConvBias::NonlineMode; |
| 356 | auto arg_p1 = get_quantized_args_with_nlmode(NLMode::IDENTITY), |
| 357 | arg_p2 = get_quantized_args_with_nlmode(NLMode::RELU), |
| 358 | arg_p3 = get_quantized_args_with_nlmode(NLMode::H_SWISH); |
| 359 | std::vector<TestArg> args; |
| 360 | args.insert(args.end(), arg_p1.begin(), arg_p1.end()); |
| 361 | args.insert(args.end(), arg_p2.begin(), arg_p2.end()); |
| 362 | args.insert(args.end(), arg_p3.begin(), arg_p3.end()); |
| 363 | return args; |
| 364 | } |
| 365 | |
| 366 | std::vector<TestArg> get_int8_nchw4_args(size_t kernel_size) { |
| 367 | std::vector<TestArg> args; |