From 35134b52d3b03ecef4d525d5df83ea9425becf84 Mon Sep 17 00:00:00 2001 From: liuchengji Date: Wed, 20 Aug 2025 20:09:59 +0800 Subject: [PATCH] fix output tensor force from fp16 to fp32 --- mindspore-lite/src/litert/lite_session.cc | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/mindspore-lite/src/litert/lite_session.cc b/mindspore-lite/src/litert/lite_session.cc index 5fc41f44..e1419e6e 100644 --- a/mindspore-lite/src/litert/lite_session.cc +++ b/mindspore-lite/src/litert/lite_session.cc @@ -454,9 +454,7 @@ int LiteSession::IsolateOutputTensor() { new_tensor->set_init_ref_count(src_tensor->init_ref_count()); /* src tensor set for graph calculate */ - if (src_tensor->data_type() == kNumberTypeFloat16) { - src_tensor->set_data_type(kNumberTypeFloat32); - } + src_tensor->set_ref_count(1); isolate_graph_output_map_.insert(std::make_pair(new_tensor, src_tensor)); -- Gitee