summaryrefslogtreecommitdiffstats
path: root/config-model/src/main/java/com/yahoo/schema/processing/OnnxModelConfigGenerator.java
diff options
context:
space:
mode:
Diffstat (limited to 'config-model/src/main/java/com/yahoo/schema/processing/OnnxModelConfigGenerator.java')
-rw-r--r--config-model/src/main/java/com/yahoo/schema/processing/OnnxModelConfigGenerator.java3
1 files changed, 3 insertions, 0 deletions
diff --git a/config-model/src/main/java/com/yahoo/schema/processing/OnnxModelConfigGenerator.java b/config-model/src/main/java/com/yahoo/schema/processing/OnnxModelConfigGenerator.java
index ce56a4320d3..338977fa679 100644
--- a/config-model/src/main/java/com/yahoo/schema/processing/OnnxModelConfigGenerator.java
+++ b/config-model/src/main/java/com/yahoo/schema/processing/OnnxModelConfigGenerator.java
@@ -50,6 +50,9 @@ public class OnnxModelConfigGenerator extends Processor {
if (profile.getSecondPhaseRanking() != null) {
process(profile.getSecondPhaseRanking().getRoot(), profile);
}
+ if (profile.getGlobalPhaseRanking() != null) {
+ process(profile.getGlobalPhaseRanking().getRoot(), profile);
+ }
for (Map.Entry<String, RankProfile.RankingExpressionFunction> function : profile.getFunctions().entrySet()) {
process(function.getValue().function().getBody().getRoot(), profile);
}