From fc732c8212ad01ba53ab98a0c7dceba64c91481d Mon Sep 17 00:00:00 2001 From: laiyongqiang Date: Fri, 25 Dec 2020 17:35:42 +0800 Subject: [PATCH] change summary input tensor to kLifeLongGraphAll --- mindspore/ccsrc/backend/optimizer/somas/somas.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/ccsrc/backend/optimizer/somas/somas.cc b/mindspore/ccsrc/backend/optimizer/somas/somas.cc index 40f7a07fa8..a0a88511cc 100644 --- a/mindspore/ccsrc/backend/optimizer/somas/somas.cc +++ b/mindspore/ccsrc/backend/optimizer/somas/somas.cc @@ -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()