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 815f68128f..56c87c8831 100644 --- a/onnxruntime/core/providers/coreml/builders/impl/split_op_builder.cc +++ b/onnxruntime/core/providers/coreml/builders/impl/split_op_builder.cc @@ -139,8 +139,8 @@ bool SplitOpBuilder::IsOpSupportedImpl(const Node& node, const OpBuilderInputPar } const auto& splits_tensor = *initializers.at(input_defs[1]->Name()); Initializer unpacked_tensor(splits_tensor); - auto splits_span = unpacked_tensor.DataAsSpan(); - int sum_of_splits = std::accumulate(splits_span.begin(), splits_span.end(), 0); + auto splits_span = unpacked_tensor.DataAsSpan(); + int64_t sum_of_splits = std::accumulate(splits_span.begin(), splits_span.end(), int64_t{0}); if (sum_of_splits != split_dims_at_axis) { LOGS(logger, VERBOSE) << "Mismatch between the sum of 'split'. Expected: " << split_dims_at_axis