提交 dda28b0e 编写于 作者: J jerrywgz

fix bug in if-else op, test=develop

上级 9035bb81
......@@ -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.
先完成此消息的编辑!
想要评论请 注册