fix onehot bug

Signed-off-by: zjun <zhangjun0@huawei.com>
This commit is contained in:
zjun 2020-11-19 19:09:50 +08:00
parent 85e4efd706
commit 78d5bcac50
1 changed files with 1 additions and 1 deletions

View File

@ -1972,8 +1972,8 @@ void PynativeExecutor::Clear(const std::string &flag) {
ConfigManager::GetInstance().ResetIterNum(); ConfigManager::GetInstance().ResetIterNum();
if (top_graph_cells_.find(flag) != top_graph_cells_.end()) { if (top_graph_cells_.find(flag) != top_graph_cells_.end()) {
op_forward_map_.clear(); op_forward_map_.clear();
Clean();
} }
Clean();
return; return;
} }