diff options
author | Arne Juul <arnej@yahooinc.com> | 2023-11-10 10:02:02 +0000 |
---|---|---|
committer | Arne Juul <arnej@yahooinc.com> | 2023-11-10 10:14:23 +0000 |
commit | 1ec05cfed625e6395b8d3346c4f15b5bc7507dcf (patch) | |
tree | cbe16302842983afd9192b78a745532e419cd04a /model-integration | |
parent | 27c62ead49c88382f128034c48e22d77a83f8104 (diff) |
unpack_bits_from_int8 -> unpack_bits
Diffstat (limited to 'model-integration')
-rw-r--r-- | model-integration/src/main/java/ai/vespa/embedding/ColBertEmbedder.java | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/model-integration/src/main/java/ai/vespa/embedding/ColBertEmbedder.java b/model-integration/src/main/java/ai/vespa/embedding/ColBertEmbedder.java index a3273abff57..d42ec629bf7 100644 --- a/model-integration/src/main/java/ai/vespa/embedding/ColBertEmbedder.java +++ b/model-integration/src/main/java/ai/vespa/embedding/ColBertEmbedder.java @@ -13,7 +13,7 @@ import com.yahoo.language.process.Embedder; import com.yahoo.searchlib.rankingexpression.evaluation.MapContext; import com.yahoo.searchlib.rankingexpression.evaluation.TensorValue; import com.yahoo.searchlib.rankingexpression.rule.ReferenceNode; -import com.yahoo.searchlib.rankingexpression.rule.UnpackBitsFromInt8; +import com.yahoo.searchlib.rankingexpression.rule.UnpackBitsNode; import com.yahoo.tensor.IndexedTensor; import com.yahoo.tensor.Tensor; import com.yahoo.tensor.TensorAddress; @@ -284,7 +284,7 @@ public class ColBertEmbedder extends AbstractComponent implements Embedder { } public static Tensor expandBitTensor(Tensor packed) { - var unpacker = new UnpackBitsFromInt8(new ReferenceNode("input"), TensorType.Value.FLOAT, "big"); + var unpacker = new UnpackBitsNode(new ReferenceNode("input"), TensorType.Value.FLOAT, "big"); var context = new MapContext(); context.put("input", new TensorValue(packed)); return unpacker.evaluate(context).asTensor(); |