diff --git a/mindspore-lite/src/litert/lite_session.cc b/mindspore-lite/src/litert/lite_session.cc index 5fc41f44fb7a9bf130469130ed846a6d5bd289ee..e1419e6e21188183cbe6d203f5ef4d79b1a9f639 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));