Package com.yahoo.schema
Class OnnxModel
java.lang.Object
com.yahoo.schema.DistributableResource
com.yahoo.schema.OnnxModel
- All Implemented Interfaces:
Comparable<DistributableResource>
A global ONNX model distributed using file distribution, similar to ranking constants.
- Author:
- lesters
-
Nested Class Summary
Nested classes/interfaces inherited from class com.yahoo.schema.DistributableResource
DistributableResource.PathType
-
Constructor Summary
-
Method Summary
Modifier and TypeMethodDescriptionvoid
addInputNameMapping
(String onnxName, String vespaName) void
addInputNameMapping
(String onnxName, String vespaName, boolean overwrite) void
addOutputNameMapping
(String onnxName, String vespaName) void
addOutputNameMapping
(String onnxName, String vespaName, boolean overwrite) void
setGpuDevice
(int deviceNumber, boolean required) void
setModelInfo
(OnnxModelInfo modelInfo) void
setStatelessExecutionMode
(String executionMode) void
setStatelessInterOpThreads
(int interOpThreads) void
setStatelessIntraOpThreads
(int intraOpThreads) void
Methods inherited from class com.yahoo.schema.DistributableResource
compareTo, getFileName, getFilePath, getFileReference, getName, getPathType, getUri, register, register, setFileName, toString, validate
-
Constructor Details
-
OnnxModel
-
OnnxModel
-
-
Method Details
-
setUri
- Overrides:
setUri
in classDistributableResource
-
addInputNameMapping
-
addInputNameMapping
-
addOutputNameMapping
-
addOutputNameMapping
-
setModelInfo
-
getInputMap
-
getOutputMap
-
getInitializers
-
getDefaultOutput
-
setStatelessExecutionMode
-
getStatelessExecutionMode
-
setStatelessInterOpThreads
public void setStatelessInterOpThreads(int interOpThreads) -
getStatelessInterOpThreads
-
setStatelessIntraOpThreads
public void setStatelessIntraOpThreads(int intraOpThreads) -
setGpuDevice
public void setGpuDevice(int deviceNumber, boolean required) -
getStatelessIntraOpThreads
-
getGpuDevice
-