Package com.yahoo.schema

Class OnnxModel

java.lang.Object

com.yahoo.schema.DistributableResource

com.yahoo.schema.OnnxModel

All Implemented Interfaces:

Comparable<DistributableResource>

public class OnnxModel
extends 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

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)
String	getDefaultOutput()
Optional<Integer>	getGpuDevice()
Map<String,String>	getInputMap()
Map<String,String>	getOutputMap()
Optional<String>	getStatelessExecutionMode()
Optional<Integer>	getStatelessInterOpThreads()
Optional<Integer>	getStatelessIntraOpThreads()
void	setGpuDevice(int deviceNumber)
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

clone, 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

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()

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)

setGpuDevice

public void setGpuDevice(int deviceNumber)

getStatelessIntraOpThreads

public Optional<Integer> getStatelessIntraOpThreads()

getGpuDevice

public Optional<Integer> getGpuDevice()