提交 e15b19ef 编写于 作者: U Unknown 提交者: liutuo

cpplint check

上级 6e1b70a3
......@@ -21,7 +21,6 @@ void DepthToSpaceOpFunctor<DeviceType::OPENCL, T>::operator()(
const index_t input_depth = input->dim(3);
const char *kernel_name = nullptr;
index_t kernel_width = input_width;
index_t output_height, output_width, output_depth;
if (d2s_) {
......@@ -29,13 +28,11 @@ void DepthToSpaceOpFunctor<DeviceType::OPENCL, T>::operator()(
output_width = input_width * block_size_;
output_depth = input_depth / (block_size_ * block_size_);
kernel_name = "depth_to_space";
kernel_width = output_width;
} else {
output_height = input_height / block_size_;
output_width = input_width / block_size_;
output_depth = input_depth * block_size_ * block_size_;
kernel_name = "space_to_depth";
kernel_width = input_width;
}
const index_t input_depth_blocks = RoundUpDiv4(input_depth);
const index_t output_depth_blocks = RoundUpDiv4(output_depth);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册