Class OnnxModelsConfig.Model.Builder
java.lang.Object
com.yahoo.vespa.config.search.core.OnnxModelsConfig.Model.Builder
- All Implemented Interfaces:
com.yahoo.config.ConfigBuilder
- Enclosing class:
- OnnxModelsConfig.Model
public static final class OnnxModelsConfig.Model.Builder
extends Object
implements com.yahoo.config.ConfigBuilder
-
Field Summary
FieldsModifier and TypeFieldDescription -
Constructor Summary
Constructors -
Method Summary
Modifier and TypeMethodDescriptionbuild()dry_run_on_setup(boolean __value) gpu_device(int __value) gpu_device_required(boolean __value) input(OnnxModelsConfig.Model.Input.Builder __builder) Add the given builder to this builder's list of Input buildersMake a new builder and run the supplied function on it before adding it to the listinput(List<OnnxModelsConfig.Model.Input.Builder> __builders) Set the given list as this builder's list of Input buildersoutput(OnnxModelsConfig.Model.Output.Builder __builder) Add the given builder to this builder's list of Output buildersMake a new builder and run the supplied function on it before adding it to the listoutput(List<OnnxModelsConfig.Model.Output.Builder> __builders) Set the given list as this builder's list of Output buildersstateless_execution_mode(String __value) stateless_interop_threads(int __value) stateless_intraop_threads(int __value)
-
Field Details
-
input
-
output
-
-
Constructor Details
-
Builder
public Builder() -
Builder
-
-
Method Details
-
name
-
fileref
-
input
Add the given builder to this builder's list of Input builders- Parameters:
__builder- a builder- Returns:
- this builder
-
input
Make a new builder and run the supplied function on it before adding it to the list- Parameters:
__func- lambda that modifies the given builder- Returns:
- this builder
-
input
Set the given list as this builder's list of Input builders- Parameters:
__builders- a list of builders- Returns:
- this builder
-
output
Add the given builder to this builder's list of Output builders- Parameters:
__builder- a builder- Returns:
- this builder
-
output
public OnnxModelsConfig.Model.Builder output(Consumer<OnnxModelsConfig.Model.Output.Builder> __func) Make a new builder and run the supplied function on it before adding it to the list- Parameters:
__func- lambda that modifies the given builder- Returns:
- this builder
-
output
public OnnxModelsConfig.Model.Builder output(List<OnnxModelsConfig.Model.Output.Builder> __builders) Set the given list as this builder's list of Output builders- Parameters:
__builders- a list of builders- Returns:
- this builder
-
dry_run_on_setup
-
stateless_execution_mode
-
stateless_interop_threads
-
stateless_intraop_threads
-
gpu_device
-
gpu_device_required
-
build
-