diff options
author | Lester Solbakken <lesters@oath.com> | 2022-01-10 14:34:04 +0100 |
---|---|---|
committer | gjoranv <gv@verizonmedia.com> | 2022-06-08 11:45:28 +0200 |
commit | 19f9c783c2f1ca136a6ed874656e0a2c93b4adca (patch) | |
tree | fe0a0ccaff7e0ba5293d9950b1725d7e03c5237d /config-model/src/main/java/com/yahoo/schema/RankProfile.java | |
parent | 45a16605fe7caa2ebcbc1068fa2b48cbfa3b28c1 (diff) |
onnxModel to onnx in summary/matchfeatures
Diffstat (limited to 'config-model/src/main/java/com/yahoo/schema/RankProfile.java')
-rw-r--r-- | config-model/src/main/java/com/yahoo/schema/RankProfile.java | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/config-model/src/main/java/com/yahoo/schema/RankProfile.java b/config-model/src/main/java/com/yahoo/schema/RankProfile.java index 5479ecf323f..56786c733ec 100644 --- a/config-model/src/main/java/com/yahoo/schema/RankProfile.java +++ b/config-model/src/main/java/com/yahoo/schema/RankProfile.java @@ -1089,11 +1089,11 @@ public class RankProfile implements Cloneable { Map<String, TensorType> inputTypes = resolveOnnxInputTypes(model, context); TensorType defaultOutputType = model.getTensorType(model.getDefaultOutput(), inputTypes); - context.setType(new Reference("onnxModel", args, null), defaultOutputType); + context.setType(new Reference("onnx", args, null), defaultOutputType); for (Map.Entry<String, String> mapping : model.getOutputMap().entrySet()) { TensorType type = model.getTensorType(mapping.getKey(), inputTypes); - context.setType(new Reference("onnxModel", args, mapping.getValue()), type); + context.setType(new Reference("onnx", args, mapping.getValue()), type); } } return context; |