Merge pull request #14956 from jerrywgz/fix_bug_in_ifelse

fix bug in if-else op

Merge pull request #14956 from jerrywgz/fix_bug_in_ifelse
fix bug in if-else op
ef2d292b · jerrywgz · GitHub · dc8eca82 · dda28b0e · ef2d292b
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

paddle/fluid/operators/split_lod_tensor_op.cc paddle/fluid/operators/split_lod_tensor_op.cc +1 -1

未找到文件。
--- 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<bool>();

-    std::vector<std::vector<CopyRange>> copy_ranges(mask_dim[0]);
+    std::vector<std::vector<CopyRange>> copy_ranges(2);

    // set out_true/out_false lod
    for (size_t t = 0; t < 2; t++) {