Package com.yahoo.schema

Class OnnxModel

java.lang.Object

com.yahoo.schema.DistributableResource

com.yahoo.schema.OnnxModel

All Implemented Interfaces:

Cloneable, Comparable<DistributableResource>

public class OnnxModel
extends DistributableResource
implements Cloneable

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

Constructors

Constructor	Description
OnnxModel(String name)
OnnxModel(String name, String fileName)

Method Summary

Modifier and Type	Method	Description
void	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)
OnnxModel	clone()
String	getDefaultOutput()
Optional<com.yahoo.config.model.api.OnnxModelOptions.GpuDevice>	getGpuDevice()
Set<String>	getInitializers()
Map<String,String>	getInputMap()
Map<String,String>	getOutputMap()
Optional<String>	getStatelessExecutionMode()
Optional<Integer>	getStatelessInterOpThreads()
Optional<Integer>	getStatelessIntraOpThreads()
com.yahoo.config.model.api.OnnxModelOptions	onnxModelOptions()
void	setGpuDevice(int deviceNumber, boolean required)
void	setModelInfo(OnnxModelInfo modelInfo)
void	setStatelessExecutionMode(String executionMode)
void	setStatelessInterOpThreads(int interOpThreads)
void	setStatelessIntraOpThreads(int intraOpThreads)
void	setUri(String uri)

Methods inherited from class com.yahoo.schema.DistributableResource

compareTo, getFileName, getFilePath, getFileReference, getName, getPathType, getUri, register, register, setFileName, toString, validate

Methods inherited from class java.lang.Object

equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

Constructor Details

OnnxModel

public OnnxModel(String name)

OnnxModel

public OnnxModel(String name,
 String fileName)

Method Details

clone

public OnnxModel clone()

Overrides:

clone in class DistributableResource

setUri

public void setUri(String uri)

Overrides:

setUri in class DistributableResource

addInputNameMapping

public void addInputNameMapping(String onnxName,
 String vespaName)

addInputNameMapping

public void addInputNameMapping(String onnxName,
 String vespaName,
 boolean overwrite)

addOutputNameMapping

public void addOutputNameMapping(String onnxName,
 String vespaName)

addOutputNameMapping

public void addOutputNameMapping(String onnxName,
 String vespaName,
 boolean overwrite)

setModelInfo

public void setModelInfo(OnnxModelInfo modelInfo)

getInputMap

public Map<String,String> getInputMap()

getOutputMap

public Map<String,String> getOutputMap()

getInitializers

public Set<String> getInitializers()

getDefaultOutput

public String getDefaultOutput()

setStatelessExecutionMode

public void setStatelessExecutionMode(String executionMode)

getStatelessExecutionMode

public Optional<String> getStatelessExecutionMode()

setStatelessInterOpThreads

public void setStatelessInterOpThreads(int interOpThreads)

getStatelessInterOpThreads

public Optional<Integer> getStatelessInterOpThreads()

setStatelessIntraOpThreads

public void setStatelessIntraOpThreads(int intraOpThreads)

getStatelessIntraOpThreads

public Optional<Integer> getStatelessIntraOpThreads()

setGpuDevice

public void setGpuDevice(int deviceNumber,
 boolean required)

getGpuDevice

public Optional<com.yahoo.config.model.api.OnnxModelOptions.GpuDevice> getGpuDevice()

onnxModelOptions

public com.yahoo.config.model.api.OnnxModelOptions onnxModelOptions()