Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
rachguo authored and rachguo committed Nov 18, 2023
1 parent 0bd59d6 commit 0d96ebe
Showing 1 changed file with 1 addition and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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<COREML_SPEC::NeuralNetworkLayer> layer = CreateNNLayer(model_builder, node);
auto* coreml_splitnd = layer->mutable_splitnd();
Expand Down Expand Up @@ -94,7 +94,6 @@ Status SplitOpBuilder::AddToModelBuilderImpl(ModelBuilder& model_builder,
}
} else {
// even
num_outputs = node.OutputDefs().size();
coreml_splitnd->set_numsplits(num_outputs);
}
}
Expand Down

0 comments on commit 0d96ebe

Please sign in to comment.