From 06358129a242c2cb58bf91c81603eccaac7b8df5 Mon Sep 17 00:00:00 2001 From: fengjiayi Date: Fri, 21 Jul 2017 13:27:34 +0800 Subject: [PATCH] Fix compile error --- paddle/operators/cross_entropy_op.cc | 2 +- paddle/operators/sgd_op.cc | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/paddle/operators/cross_entropy_op.cc b/paddle/operators/cross_entropy_op.cc index fe669b03ca..7d7bb09f3d 100644 --- a/paddle/operators/cross_entropy_op.cc +++ b/paddle/operators/cross_entropy_op.cc @@ -35,7 +35,7 @@ protected: PADDLE_ENFORCE(inputs[0]->dims().size() == 2, "X's dimension must be 2."); PADDLE_ENFORCE(outputs[0]->dims().size() == 1, "label's dimension must be 1."); - outputs[0]->set_dims(framework::make_ddim({inputs[0]->dims()[0]})); + outputs[0]->Resize(framework::make_ddim({inputs[0]->dims()[0]})); } }; diff --git a/paddle/operators/sgd_op.cc b/paddle/operators/sgd_op.cc index 04df87a3ad..66ab1e0011 100644 --- a/paddle/operators/sgd_op.cc +++ b/paddle/operators/sgd_op.cc @@ -31,7 +31,7 @@ protected: PADDLE_ENFORCE(outputs[0] != nullptr, "outputs[0] mast be set"); PADDLE_ENFORCE(inputs[0]->dims() == inputs[1]->dims(), "Two input of SGD Op's dimension must be same."); - outputs[0]->set_dims(inputs[0]->dims()); + outputs[0]->Resize(inputs[0]->dims()); } }; -- GitLab