model-integration/src/main/java/ai/vespa/rankingexpression/importer/onnx/ImportedOnnxModel.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25

// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package ai.vespa.rankingexpression.importer.onnx;

import ai.vespa.rankingexpression.importer.ImportedModel;
import onnx.Onnx;

public class ImportedOnnxModel extends ImportedModel {

    private final Onnx.ModelProto modelProto;

    public ImportedOnnxModel(String name, String source, Onnx.ModelProto modelProto) {
        super(name, source, ModelType.ONNX);
        this.modelProto = modelProto;
    }

    @Override
    public boolean isNative() {
        return false;
    }

    @Override
    public ImportedModel asNative() {
        return OnnxImporter.convertModel(name(), source(), modelProto, ModelType.ONNX);
    }
}