From 945795a46328b12e147c6899054ecd1bce9f920e Mon Sep 17 00:00:00 2001
From: Henning Baldersheim <balder@yahoo-inc.com>
Date: Wed, 17 Jan 2024 18:35:55 +0100
Subject: Allocate a larger hashtable to avoid expensive resize and associated
 equals/hash method calls.

---
 vespajlib/src/main/java/com/yahoo/tensor/functions/Reduce.java | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/vespajlib/src/main/java/com/yahoo/tensor/functions/Reduce.java b/vespajlib/src/main/java/com/yahoo/tensor/functions/Reduce.java
index 4b60af99ecb..5327457a438 100644
--- a/vespajlib/src/main/java/com/yahoo/tensor/functions/Reduce.java
+++ b/vespajlib/src/main/java/com/yahoo/tensor/functions/Reduce.java
@@ -144,7 +144,7 @@ public class Reduce<NAMETYPE extends Name> extends PrimitiveTensorFunction<NAMET
     }
 
     private static TensorAddress reduceDimensions(TensorAddress address, TensorType argumentType, TensorType reducedType, List<String> dimensions) {
-        Set<Integer> indexesToRemove = new HashSet<>();
+        Set<Integer> indexesToRemove = new HashSet<>(dimensions.size()*2);
         for (String dimensionToRemove : dimensions)
             indexesToRemove.add(argumentType.indexOfDimension(dimensionToRemove).get());
 
-- 
cgit v1.2.3