From 0d96ebe4e165d138bedb939f0a4ec344f37ad693 Mon Sep 17 00:00:00 2001 From: rachguo Date: Fri, 17 Nov 2023 17:29:39 -0800 Subject: [PATCH] update --- .../core/providers/coreml/builders/impl/split_op_builder.cc | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/onnxruntime/core/providers/coreml/builders/impl/split_op_builder.cc b/onnxruntime/core/providers/coreml/builders/impl/split_op_builder.cc index 9448e94699e4e..1ccc9e150da96 100644 --- a/onnxruntime/core/providers/coreml/builders/impl/split_op_builder.cc +++ b/onnxruntime/core/providers/coreml/builders/impl/split_op_builder.cc @@ -61,7 +61,7 @@ Status SplitOpBuilder::AddToModelBuilderImpl(ModelBuilder& model_builder, const auto axis = helper.Get("axis", 0); // attribute introduced since opset 18 - uint64_t num_outputs = 2; + uint64_t num_outputs; std::unique_ptr layer = CreateNNLayer(model_builder, node); auto* coreml_splitnd = layer->mutable_splitnd(); @@ -94,7 +94,6 @@ Status SplitOpBuilder::AddToModelBuilderImpl(ModelBuilder& model_builder, } } else { // even - num_outputs = node.OutputDefs().size(); coreml_splitnd->set_numsplits(num_outputs); } }