From 305db31301e97e940f42f6c9642f6d1f0aebc9bc Mon Sep 17 00:00:00 2001 From: Rachel Guo <35738743+YUNQIUGUO@users.noreply.github.com> Date: Thu, 7 Dec 2023 14:48:55 -0800 Subject: [PATCH] fix build aar error in Zip-Nuget-Java-Nodejs Packaging pipeline (#18745) ### Description [Pipeline failure info](https://aiinfra.visualstudio.com/Lotus/_build/results?buildId=387310&view=logs&j=0aae05c9-1dc0-5099-eb4a-4cbb949c7458&t=71450a55-3e84-511c-7394-a06145376912&l=1044) ### Motivation and Context Fix packaging pipeline brought by pr. Co-authored-by: rachguo --- .../nnapi/nnapi_builtin/builders/impl/split_op_builder.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/onnxruntime/core/providers/nnapi/nnapi_builtin/builders/impl/split_op_builder.cc b/onnxruntime/core/providers/nnapi/nnapi_builtin/builders/impl/split_op_builder.cc index 4aef9f0d27..68b63badb8 100644 --- a/onnxruntime/core/providers/nnapi/nnapi_builtin/builders/impl/split_op_builder.cc +++ b/onnxruntime/core/providers/nnapi/nnapi_builtin/builders/impl/split_op_builder.cc @@ -95,7 +95,7 @@ bool SplitOpBuilder::IsOpSupportedImpl(const InitializedTensorSet& initializers, NodeAttrHelper helper(node_unit); const auto axis = helper.Get("axis", 0); - const auto split_dims_at_axis = input_shape[HandleNegativeAxis(axis, input_shape.size())]; + const auto split_dims_at_axis = input_shape[SafeInt(HandleNegativeAxis(axis, input_shape.size()))]; if (input_defs.size() > 1 && input_defs[1].node_arg.Exists()) { // if optional input `split` is provided auto split_initializer_it = initializers.find(input_defs[1].node_arg.Name());