From f72c1501c5dc48e45e8b56bb6ccec7724993f08c Mon Sep 17 00:00:00 2001 From: ybrnathan <7902510+ybrnathan@users.noreply.github.com> Date: Mon, 10 Jun 2019 12:54:17 -0700 Subject: [PATCH] Optimize concat PrepareForCompute perf by reserve vector size. (#1198) --- onnxruntime/core/providers/cpu/tensor/concat.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/onnxruntime/core/providers/cpu/tensor/concat.cc b/onnxruntime/core/providers/cpu/tensor/concat.cc index 06b7bd9a47..b970270509 100644 --- a/onnxruntime/core/providers/cpu/tensor/concat.cc +++ b/onnxruntime/core/providers/cpu/tensor/concat.cc @@ -80,6 +80,7 @@ Status ConcatBase::PrepareForCompute(OpKernelContext* ctx, int input_count, Prep p.output_axis_pitch = 1; for (size_t i = inputs_0_rank; i-- > axis;) p.output_axis_pitch *= dims[i]; + p.inputs.reserve(input_count); for (int input_index = 0; input_index < input_count; input_index++) { const Tensor* data_n_ptr = ctx->Input(input_index); auto& data_n = *data_n_ptr;