未验证 提交 ef2d292b 编写于 作者: J jerrywgz 提交者: GitHub

Merge pull request #14956 from jerrywgz/fix_bug_in_ifelse

fix bug in if-else op
......@@ -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++) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册