Package com.yahoo.embedding

Class BertBaseEmbedderConfig.Builder

java.lang.Object

com.yahoo.embedding.BertBaseEmbedderConfig.Builder

All Implemented Interfaces:

com.yahoo.config.ConfigBuilder, com.yahoo.config.ConfigInstance.Builder

Enclosing class:

BertBaseEmbedderConfig

public static final class BertBaseEmbedderConfig.Builder
extends Object
implements com.yahoo.config.ConfigInstance.Builder

Constructor Summary

Constructors

Constructor	Description
Builder()
Builder(BertBaseEmbedderConfig config)

Method Summary

Modifier and Type	Method	Description
BertBaseEmbedderConfig	build()
final boolean	dispatchGetConfig(com.yahoo.config.ConfigInstance.Producer producer)
final boolean	getApplyOnRestart()
final String	getDefMd5()
final String	getDefName()
final String	getDefNamespace()
BertBaseEmbedderConfig.Builder	onnxExecutionMode(BertBaseEmbedderConfig.OnnxExecutionMode.Enum __value)
BertBaseEmbedderConfig.Builder	onnxGpuDevice(int __value)
BertBaseEmbedderConfig.Builder	onnxInterOpThreads(int __value)
BertBaseEmbedderConfig.Builder	onnxIntraOpThreads(int __value)
BertBaseEmbedderConfig.Builder	poolingStrategy(BertBaseEmbedderConfig.PoolingStrategy.Enum __value)
final void	setApplyOnRestart(boolean applyOnRestart)
BertBaseEmbedderConfig.Builder	tokenizerVocab(com.yahoo.config.ModelReference __value)
BertBaseEmbedderConfig.Builder	transformerAttentionMask(String __value)
BertBaseEmbedderConfig.Builder	transformerEndSequenceToken(int __value)
BertBaseEmbedderConfig.Builder	transformerInputIds(String __value)
BertBaseEmbedderConfig.Builder	transformerMaxTokens(int __value)
BertBaseEmbedderConfig.Builder	transformerModel(com.yahoo.config.ModelReference __value)
BertBaseEmbedderConfig.Builder	transformerOutput(String __value)
BertBaseEmbedderConfig.Builder	transformerStartSequenceToken(int __value)
BertBaseEmbedderConfig.Builder	transformerTokenTypeIds(String __value)

Methods inherited from class java.lang.Object

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

Constructor Details

Builder

public Builder()

Builder

public Builder(BertBaseEmbedderConfig config)

Method Details

tokenizerVocab

public BertBaseEmbedderConfig.Builder tokenizerVocab(com.yahoo.config.ModelReference __value)

transformerModel

public BertBaseEmbedderConfig.Builder transformerModel(com.yahoo.config.ModelReference __value)

transformerMaxTokens

public BertBaseEmbedderConfig.Builder transformerMaxTokens(int __value)

poolingStrategy

public BertBaseEmbedderConfig.Builder poolingStrategy(BertBaseEmbedderConfig.PoolingStrategy.Enum __value)

transformerInputIds

public BertBaseEmbedderConfig.Builder transformerInputIds(String __value)

transformerAttentionMask

public BertBaseEmbedderConfig.Builder transformerAttentionMask(String __value)

transformerTokenTypeIds

public BertBaseEmbedderConfig.Builder transformerTokenTypeIds(String __value)

transformerStartSequenceToken

public BertBaseEmbedderConfig.Builder transformerStartSequenceToken(int __value)

transformerEndSequenceToken

public BertBaseEmbedderConfig.Builder transformerEndSequenceToken(int __value)

transformerOutput

public BertBaseEmbedderConfig.Builder transformerOutput(String __value)

onnxExecutionMode

public BertBaseEmbedderConfig.Builder onnxExecutionMode(BertBaseEmbedderConfig.OnnxExecutionMode.Enum __value)

onnxInterOpThreads

public BertBaseEmbedderConfig.Builder onnxInterOpThreads(int __value)

onnxIntraOpThreads

public BertBaseEmbedderConfig.Builder onnxIntraOpThreads(int __value)

onnxGpuDevice

public BertBaseEmbedderConfig.Builder onnxGpuDevice(int __value)

dispatchGetConfig

public final boolean dispatchGetConfig(com.yahoo.config.ConfigInstance.Producer producer)

Specified by:

dispatchGetConfig in interface com.yahoo.config.ConfigInstance.Builder

getDefMd5

public final String getDefMd5()

Specified by:

getDefMd5 in interface com.yahoo.config.ConfigInstance.Builder

getDefName

public final String getDefName()

Specified by:

getDefName in interface com.yahoo.config.ConfigInstance.Builder

getDefNamespace

public final String getDefNamespace()

Specified by:

getDefNamespace in interface com.yahoo.config.ConfigInstance.Builder

getApplyOnRestart

public final boolean getApplyOnRestart()

Specified by:

getApplyOnRestart in interface com.yahoo.config.ConfigInstance.Builder

setApplyOnRestart

public final void setApplyOnRestart(boolean applyOnRestart)

Specified by:

setApplyOnRestart in interface com.yahoo.config.ConfigInstance.Builder

build

public BertBaseEmbedderConfig build()