| 257 | } |
| 258 | |
| 259 | void Shaper::SpaceToBatch(const std::string &input_name, |
| 260 | const std::vector<int32_t> &block_sizes, |
| 261 | const std::vector<int32_t> &pads, |
| 262 | const std::string &output_name) { |
| 263 | auto input_dimen = shape_map_.at(input_name); |
| 264 | auto output_dimen = {input_dimen[0] * Product(block_sizes), |
| 265 | (input_dimen[1] + pads[0] + pads[1]) / block_sizes[0], |
| 266 | (input_dimen[2] + pads[2] + pads[3]) / block_sizes[1], |
| 267 | input_dimen[3]}; |
| 268 | shape_map_[output_name] = output_dimen; |
| 269 | } |
| 270 | |
| 271 | void Shaper::Affine(const std::string &input_name, |
| 272 | const std::string &output_name) { |
no test coverage detected