| 16 | UniformIntRNG rng{INT8_MIN >> 1, INT8_MAX >> 1}; |
| 17 | checker.set_rng(0, &rng); |
| 18 | struct Config { |
| 19 | Param param; |
| 20 | DType dtype; |
| 21 | TensorShape shape; |
| 22 | Config(Param param, DType dtype, TensorShape shape) |
| 23 | : param(param), dtype(dtype), shape(shape) {} |
| 24 | }; |
| 25 | std::vector<Config> configs; |
| 26 | for (auto mode : {Mode::MEAN, Mode::MAX, Mode::MIN}) |
| 27 | for (auto dtype : std::vector<DType>{ |