scatter_op bug fix, test=develop (#16866)

* scatter_op bug fix, test=develop

* scatter_op bug fix, test=develop

* scatter_op bug fix, test=develop
revert-16839-cmakelist_change
liuwei1031 6 years ago committed by GitHub
parent 46bd853c10
commit 6864370a9e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -42,10 +42,6 @@ class ScatterOp : public framework::OperatorWithKernel {
PADDLE_ENFORCE_EQ(ctx->GetInputDim("Updates")[0],
ctx->GetInputDim("Ids")[0],
"Updates and Ids should have same batch-size.");
framework::DDim data_dim(updates_dims);
for (int i = 1; i < data_dim.size(); ++i) {
PADDLE_ENFORCE_EQ(data_dim[i], updates_dims[i]);
}
ctx->SetOutputDim("Out", ref_dims);
}

Loading…
Cancel
Save