Skip to content

Mace
Mace

慢慢CG / Mace
与 Fork 源项目一致

Fork自 Xiaomi / Mace

通知 1
Star 0
Fork 0
Mace
Mace
提交 60bf45cf 编写于 作者: L liuqi
浏览文件
操作

Fix some bad code style.

上级 9ebc2ecb
隐藏空白更改
内联 并排
Showing with 12 addition and 6 deletion
mace/kernels/conv_pool_2d_util.cc
浏览文件 @ 60bf45cf
...@@ -39,13 +39,16 @@ void CalcPaddingAndOutputSize(const index_t *input_shape, // NCHW ...@@ -39,13 +39,16 @@ void CalcPaddingAndOutputSize(const index_t *input_shape, // NCHW
index_t k_extent_width = (kernel_width - 1) * dilations[1] + 1; index_t k_extent_width = (kernel_width - 1) * dilations[1] + 1;
switch (padding) { switch (padding) {
case VALID:output_height = (input_shape[2] - k_extent_height) / strides[0] + 1; case VALID:
output_height = (input_shape[2] - k_extent_height) / strides[0] + 1;
output_width = (input_shape[3] - k_extent_width) / strides[1] + 1; output_width = (input_shape[3] - k_extent_width) / strides[1] + 1;
break; break;
case SAME:output_height = (input_shape[2] - 1) / strides[0] + 1; case SAME:
output_height = (input_shape[2] - 1) / strides[0] + 1;
output_width = (input_shape[3] - 1) / strides[1] + 1; output_width = (input_shape[3] - 1) / strides[1] + 1;
break; break;
case FULL:output_height = (input_shape[2] + k_extent_height - 2) / strides[0] + 1; case FULL:
output_height = (input_shape[2] + k_extent_height - 2) / strides[0] + 1;
output_width = (input_shape[3] + k_extent_width - 2) / strides[1] + 1; output_width = (input_shape[3] + k_extent_width - 2) / strides[1] + 1;
break; break;
default:MACE_CHECK(false, "Unsupported padding type: ", padding); default:MACE_CHECK(false, "Unsupported padding type: ", padding);
...@@ -87,13 +90,16 @@ void CalPaddingSize(const index_t *input_shape, // NCHW ...@@ -87,13 +90,16 @@ void CalPaddingSize(const index_t *input_shape, // NCHW
index_t k_extent_width = (filter_shape[3] - 1) * dilations[1] + 1; index_t k_extent_width = (filter_shape[3] - 1) * dilations[1] + 1;
switch (padding) { switch (padding) {
case VALID:output_height = (input_shape[2] - k_extent_height) / strides[0] + 1; case VALID:
output_height = (input_shape[2] - k_extent_height) / strides[0] + 1;
output_width = (input_shape[3] - k_extent_width) / strides[1] + 1; output_width = (input_shape[3] - k_extent_width) / strides[1] + 1;
break; break;
case SAME:output_height = (input_shape[2] - 1) / strides[0] + 1; case SAME:
output_height = (input_shape[2] - 1) / strides[0] + 1;
output_width = (input_shape[3] - 1) / strides[1] + 1; output_width = (input_shape[3] - 1) / strides[1] + 1;
break; break;
case FULL:output_height = (input_shape[2] + k_extent_height - 2) / strides[0] + 1; case FULL:
output_height = (input_shape[2] + k_extent_height - 2) / strides[0] + 1;
output_width = (input_shape[3] + k_extent_width - 2) / strides[1] + 1; output_width = (input_shape[3] + k_extent_width - 2) / strides[1] + 1;
break; break;
default:MACE_CHECK(false, "Unsupported padding type: ", padding); default:MACE_CHECK(false, "Unsupported padding type: ", padding);
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或