Skip to content

Commit 8499559

Browse files
authored
"fix style" (#12600)
1 parent 445ca3d commit 8499559

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

paddle/fluid/operators/elementwise_op_function.h

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -534,8 +534,8 @@ void ElemwiseGradCompute(const framework::ExecutionContext& ctx,
534534
const framework::Tensor& dout, int axis,
535535
framework::Tensor* dx, framework::Tensor* dy,
536536
DX_OP dx_op, DY_OP dy_op) {
537-
const framework::DDim x_dim = x.dims();
538-
const framework::DDim y_dim = y.dims();
537+
const framework::DDim& x_dim = x.dims();
538+
const framework::DDim& y_dim = y.dims();
539539
if (x.dims() == y.dims()) {
540540
ElemwiseGradComputeNoBroadcast<DeviceContext, T, DX_OP, DY_OP>(
541541
ctx, x_dim, y_dim, x, y, out, dout, axis, dx, dy, dx_op, dy_op);
@@ -558,19 +558,19 @@ void ElemwiseExplicitGradCompute(const framework::ExecutionContext& ctx,
558558
framework::Tensor* dx, framework::Tensor* dy,
559559
DX_OP dx_op, DY_OP dy_op) {
560560
if (dy == nullptr) {
561-
const framework::DDim dx_dims = dout.dims();
561+
const framework::DDim& dx_dims = dout.dims();
562562
auto dy_dims = dx_dims;
563563
ElemwiseGradComputeNoBroadcast<DeviceContext, T, DX_OP, DY_OP>(
564564
ctx, dx_dims, dy_dims, x, y, out, dout, axis, dx, dy, dx_op, dy_op);
565565
} else {
566566
if (dout.dims() == dy->dims()) {
567-
const framework::DDim dx_dims = dout.dims();
568-
const framework::DDim dy_dims = dy->dims();
567+
const framework::DDim& dx_dims = dout.dims();
568+
const framework::DDim& dy_dims = dy->dims();
569569
ElemwiseGradComputeNoBroadcast<DeviceContext, T, DX_OP, DY_OP>(
570570
ctx, dx_dims, dy_dims, x, y, out, dout, axis, dx, dy, dx_op, dy_op);
571571
} else { // Y is a scalar
572572
auto dx_dims = dout.dims();
573-
const framework::DDim dy_dims = dy->dims();
573+
const framework::DDim& dy_dims = dy->dims();
574574
ElemwiseGradComputeWithBroadcast<DeviceContext, T, DX_OP, DY_OP>(
575575
ctx, dx_dims, dy_dims, x, y, out, dout, axis, dx, dy, dx_op, dy_op);
576576
}

0 commit comments

Comments
 (0)