From 8788963438730ec6efea9f9dd12287e648f6cb4a Mon Sep 17 00:00:00 2001 From: sangoly Date: Wed, 26 Jun 2019 05:47:46 +0000 Subject: [PATCH] bind tranpose output XShape for caffe2paddle model test=develop --- paddle/fluid/lite/api/paddle_use_ops.h | 2 ++ paddle/fluid/lite/kernels/arm/transpose_compute.cc | 1 + 2 files changed, 3 insertions(+) diff --git a/paddle/fluid/lite/api/paddle_use_ops.h b/paddle/fluid/lite/api/paddle_use_ops.h index cfb08bd57d..47ba27579e 100644 --- a/paddle/fluid/lite/api/paddle_use_ops.h +++ b/paddle/fluid/lite/api/paddle_use_ops.h @@ -36,3 +36,5 @@ USE_LITE_OP(depthwise_conv2d) USE_LITE_OP(pool2d) USE_LITE_OP(batch_norm) USE_LITE_OP(fusion_elementwise_sub_activation) +USE_LITE_OP(transpose) +USE_LITE_OP(transpose2) diff --git a/paddle/fluid/lite/kernels/arm/transpose_compute.cc b/paddle/fluid/lite/kernels/arm/transpose_compute.cc index 368716c368..ff32ac242b 100644 --- a/paddle/fluid/lite/kernels/arm/transpose_compute.cc +++ b/paddle/fluid/lite/kernels/arm/transpose_compute.cc @@ -170,4 +170,5 @@ REGISTER_LITE_KERNEL(transpose2, kARM, kFloat, kNCHW, paddle::lite::kernels::arm::Transpose2Compute, def) .BindInput("X", {LiteType::GetTensorTy(TARGET(kARM))}) .BindOutput("Out", {LiteType::GetTensorTy(TARGET(kARM))}) + .BindOutput("XShape", {LiteType::GetTensorTy(TARGET(kARM))}) .Finalize(); -- GitLab