Add missing reshape found by GM

paddle/fluid/operators/dequantize_op.cc

@@ -35,6 +35,13 @@ using platform::GetMKLDNNFormat;
 template <typename DeviceContext, typename T>
 class DeQuantOpKernel : public framework::OpKernel<T> {
  public:
+  void InferShape(framework::InferShapeContext *ctx) const override {
+    PADDLE_ENFORCE(ctx->HasInput("Input"), "Input should not be null");
+    PADDLE_ENFORCE(ctx->HasOutput("Output"), "Output should not be null");
+
+    ctx->SetOutputDim("Output", ctx->GetInputDim("Input"));
+    ctx->ShareLoD("Input", /*->*/ "Output");
+  }
 
   void Compute(const framework::ExecutionContext& ctx) const override {
     auto* input = ctx.Input<Tensor>("Input");

paddle/fluid/operators/quantize_op.cc

@@ -33,6 +33,14 @@ using platform::GetMKLDNNFormat;
 template <typename DeviceContext, typename T>
 class QuantOpKernel : public framework::OpKernel<T> {
  public:
+  void InferShape(framework::InferShapeContext *ctx) const override {
+    PADDLE_ENFORCE(ctx->HasInput("Input"), "Input should not be null");
+    PADDLE_ENFORCE(ctx->HasOutput("Output"), "Output should not be null");
+
+    ctx->SetOutputDim("Output", ctx->GetInputDim("Input"));
+    ctx->ShareLoD("Input", /*->*/ "Output");
+  }
+
   void Compute(const framework::ExecutionContext& ctx) const override {
     auto* input = ctx.Input<Tensor>("Input");
     auto* scale = ctx.Input<Tensor>("Scale");

paddle/fluid/operators/requantize_op.cc

@@ -34,6 +34,14 @@ using platform::GetMKLDNNFormat;
 template <typename DeviceContext, typename T>
 class ReQuantOpKernel : public framework::OpKernel<T> {
  public:
+  void InferShape(framework::InferShapeContext *ctx) const override {
+    PADDLE_ENFORCE(ctx->HasInput("Input"), "Input should not be null");
+    PADDLE_ENFORCE(ctx->HasOutput("Output"), "Output should not be null");
+
+    ctx->SetOutputDim("Output", ctx->GetInputDim("Input"));
+    ctx->ShareLoD("Input", /*->*/ "Output");
+  }
+
   void Compute(const framework::ExecutionContext& ctx) const override {
     auto* input = ctx.Input<Tensor>("Input");
     auto* scale = ctx.Input<Tensor>("Scale");