From dda28b0e682859c3868efe1ce65d636363faafd6 Mon Sep 17 00:00:00 2001 From: jerrywgz Date: Wed, 19 Dec 2018 06:50:10 +0000 Subject: [PATCH] fix bug in if-else op, test=develop --- paddle/fluid/operators/split_lod_tensor_op.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddle/fluid/operators/split_lod_tensor_op.cc b/paddle/fluid/operators/split_lod_tensor_op.cc index 767449cde98..5ede972c71f 100644 --- a/paddle/fluid/operators/split_lod_tensor_op.cc +++ b/paddle/fluid/operators/split_lod_tensor_op.cc @@ -63,7 +63,7 @@ class SplitLoDTensorOp : public framework::OperatorBase { } auto *mask_data = cpu_mask->data(); - std::vector> copy_ranges(mask_dim[0]); + std::vector> copy_ranges(2); // set out_true/out_false lod for (size_t t = 0; t < 2; t++) { -- GitLab