From 2619bdad45ab1081f6b040e342f869e398991260 Mon Sep 17 00:00:00 2001 From: seiriosPlus Date: Sun, 27 Sep 2020 23:08:14 +0800 Subject: [PATCH] merge develop --- paddle/fluid/operators/distributed_ops/split_byref_op.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/paddle/fluid/operators/distributed_ops/split_byref_op.cc b/paddle/fluid/operators/distributed_ops/split_byref_op.cc index 5dc46ba423..5d26c80f88 100644 --- a/paddle/fluid/operators/distributed_ops/split_byref_op.cc +++ b/paddle/fluid/operators/distributed_ops/split_byref_op.cc @@ -24,8 +24,8 @@ class SplitByrefOp : public framework::OperatorWithKernel { using framework::OperatorWithKernel::OperatorWithKernel; void InferShape(framework::InferShapeContext *ctx) const override { - OP_INOUT_CHECK(ctx->HasInput("X"), "Input", "Ids", "SplitOp"); - OP_INOUT_CHECK(ctx->HasOutput("Out"), "Output", "Out", "SplitOp"); + OP_INOUT_CHECK(ctx->HasInput("X"), "Input", "Ids", "SplitByrefOp"); + OP_INOUT_CHECK(ctx->HasOutputs("Out"), "Output", "Out", "SplitByrefOp"); auto in_dims = ctx->GetInputDim("X"); auto outs_names = ctx->Outputs("Out"); -- GitLab