Browse Source

!10543 Change summary input tensor to kLifeLongGraphAll

From: @laiyongqiang
Reviewed-by: @zhoufeng54,@kisnwang
Signed-off-by: @kisnwang
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 5 years ago
parent
commit
0029eb9299
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ccsrc/backend/optimizer/somas/somas.cc

+ 1
- 1
mindspore/ccsrc/backend/optimizer/somas/somas.cc View File

@@ -334,7 +334,7 @@ void Somas::SummaryInputProcess(const session::KernelGraph *graph) {
auto input_node = iter->second;
if (index < input_node->output_tensors_.size()) {
auto tensor = iter->second->output_tensors_[index];
tensor->lifelong_value_ = kLifeLongGraphEnd;
tensor->lifelong_value_ = kLifeLongGraphAll;
tensor->type_ = kSummaryInput;
total_summary_size += tensor->GetAlignedSize();
MS_LOG(INFO) << "Set summary node input tensor's lifelong, node: " << node->fullname_with_scope()


Loading…
Cancel
Save