Skip to content

Commit 608746a

Browse files
committed
use reference also for proto_init
1 parent f27ec2d commit 608746a

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

onnxruntime/core/providers/openvino/backend_manager.cc

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -541,15 +541,15 @@ BackendManager::GetModelProtoFromFusedNode(const onnxruntime::Node& fused_node,
541541
if (it == proto_initializer_map.end())
542542
continue;
543543

544-
auto* proto_init = it->second;
545-
546-
if (!proto_init) {
544+
if (!it->second) {
547545
ORT_THROW(name + " proto initializer is null!");
548546
}
549547

548+
auto& proto_init = *it->second;
549+
550550
// If the proto initializer is missing data, fill it in
551-
if (!proto_init->has_raw_data() && src_init->has_raw_data()) {
552-
*proto_init->mutable_raw_data() = src_init->raw_data();
551+
if (!proto_init.has_raw_data() && src_init->has_raw_data()) {
552+
*(proto_init.mutable_raw_data()) = src_init->raw_data();
553553
}
554554

555555
// Only set in-memory external_data fields if the data is in memory
@@ -559,7 +559,7 @@ BackendManager::GetModelProtoFromFusedNode(const onnxruntime::Node& fused_node,
559559
<< ", data_type: " << src_init->data_type()
560560
<< ", raw_data size: " << src_init->raw_data().size();
561561
if (src_init->raw_data().size() > 0) {
562-
SetExternalDataFields(*proto_init, src_init->raw_data().data(), src_init->raw_data().size());
562+
SetExternalDataFields(proto_init, src_init->raw_data().data(), src_init->raw_data().size());
563563
}
564564
else {
565565
LOGS(logger, VERBOSE) << "Initializer has empty raw_data: skipping initializer '" << src_init->name() << "'...";
@@ -576,7 +576,7 @@ BackendManager::GetModelProtoFromFusedNode(const onnxruntime::Node& fused_node,
576576

577577
LOGS(logger, VERBOSE) << "In-memory initializer EXT: " << src_init->name() << ", size: " << length;
578578

579-
SetExternalDataFields(*proto_init, (const void*)offset, length);
579+
SetExternalDataFields(proto_init, (const void*)offset, length);
580580
} else {
581581
LOGS(logger, VERBOSE) << "File-based initializer: " << src_init->name() << ", data_type: " << src_init->data_type();
582582
}

0 commit comments

Comments
 (0)