提交 4ffb73fd 编写于 作者: S sweetsky0901

format ..

上级 5b449b60
...@@ -21,8 +21,7 @@ namespace math { ...@@ -21,8 +21,7 @@ namespace math {
template <typename T> template <typename T>
__global__ void KernelUnpool2dMax(const int nthreads, const T* input_data, __global__ void KernelUnpool2dMax(const int nthreads, const T* input_data,
const int* indices_data, const int* indices_data,
const int input_height, const int input_height, const int input_width,
const int input_width,
const int channels, T* output_data, const int channels, T* output_data,
const int output_height, const int output_height,
const int output_width) { const int output_width) {
......
...@@ -94,12 +94,11 @@ class UnpoolOp : public framework::OperatorWithKernel { ...@@ -94,12 +94,11 @@ class UnpoolOp : public framework::OperatorWithKernel {
"Output(Out) of UnpoolOp should not be null."); "Output(Out) of UnpoolOp should not be null.");
auto in_x_dims = ctx->GetInputDim("X"); auto in_x_dims = ctx->GetInputDim("X");
auto in_y_dims = ctx->GetInputDim("Indices"); auto in_y_dims = ctx->GetInputDim("Indices");
std::string unpooling_type = ctx->Attrs() std::string unpooling_type =
.Get<std::string>("unpooling_type"); ctx->Attrs().Get<std::string>("unpooling_type");
std::vector<int> ksize = ctx->Attrs().Get<std::vector<int>>("ksize"); std::vector<int> ksize = ctx->Attrs().Get<std::vector<int>>("ksize");
std::vector<int> strides = ctx->Attrs().Get<std::vector<int>>("strides"); std::vector<int> strides = ctx->Attrs().Get<std::vector<int>>("strides");
std::vector<int> paddings = std::vector<int> paddings = ctx->Attrs().Get<std::vector<int>>("paddings");
ctx->Attrs().Get<std::vector<int>>("paddings");
PADDLE_ENFORCE(in_x_dims.size() == 4, PADDLE_ENFORCE(in_x_dims.size() == 4,
"Unpooling intput must be of 4-dimensional."); "Unpooling intput must be of 4-dimensional.");
PADDLE_ENFORCE_EQ(in_x_dims, in_y_dims); PADDLE_ENFORCE_EQ(in_x_dims, in_y_dims);
...@@ -142,4 +141,3 @@ REGISTER_OP_CPU_KERNEL(unpool, ...@@ -142,4 +141,3 @@ REGISTER_OP_CPU_KERNEL(unpool,
REGISTER_OP_CPU_KERNEL( REGISTER_OP_CPU_KERNEL(
unpool_grad, ops::UnpoolGradKernel<paddle::platform::CPUPlace, float>, unpool_grad, ops::UnpoolGradKernel<paddle::platform::CPUPlace, float>,
ops::UnpoolGradKernel<paddle::platform::CPUPlace, double>); ops::UnpoolGradKernel<paddle::platform::CPUPlace, double>);
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册