33 assert(
input()->shape().num_dims() == 4);
34 assert(
output()->element_type() == DataType::FLOAT32 ||
35 output()->element_type() == DataType::U8 ||
output()->element_type() == DataType::S8 ||
36 output()->element_type() == DataType::S32 ||
output()->element_type() == DataType::S64);
37 assert(
input()->element_type() ==
output()->element_type());
42 int32_t output_height = input_height / block_size;
43 int32_t output_width = input_width / block_size;
45 assert(input_height == output_height * block_size);
46 assert(input_width == output_width * block_size);