Uses of Class
org.tribuo.util.onnx.ONNXContext
Package
Description
Provides the core interfaces and classes for using Tribuo.
Interfaces and utilities for writing ONNX models from Java.
-
Uses of ONNXContext in org.tribuo
Modifier and TypeMethodDescriptionstatic <M extends com.oracle.labs.mlrg.olcut.provenance.Provenancable<ModelProvenance>>
OnnxMl.ModelProtoONNXExportable.buildModel
(ONNXContext onnxContext, String domain, long modelVersion, M model) Creates an ONNX model protobuf for the supplied context. -
Uses of ONNXContext in org.tribuo.math.onnx
Modifier and TypeMethodDescriptionstatic ONNXInitializer
ONNXMathUtils.floatMatrix
(ONNXContext context, String name, Matrix matrix, boolean transpose) Builds aONNXInitializer
containing theMatrix
.static ONNXInitializer
ONNXMathUtils.floatVector
(ONNXContext context, String name, SGDVector vector) Builds aONNXInitializer
containing theSGDVector
. -
Uses of ONNXContext in org.tribuo.util.onnx
Modifier and TypeMethodDescriptionstatic OnnxMl.TensorProto
ONNXUtils.arrayBuilder
(ONNXContext context, String name, double[] parameters) Builds a TensorProto containing the array.static OnnxMl.TensorProto
ONNXUtils.arrayBuilder
(ONNXContext context, String name, double[] parameters, boolean downcast) Builds a TensorProto containing the array.static OnnxMl.TensorProto
ONNXUtils.arrayBuilder
(ONNXContext context, String name, float[] parameters) Builds a TensorProto containing the array.static OnnxMl.TensorProto
ONNXUtils.arrayBuilder
(ONNXContext context, String name, int[] parameters) Builds a TensorProto containing the array.static OnnxMl.TensorProto
ONNXUtils.arrayBuilder
(ONNXContext context, String name, long[] parameters) Builds a TensorProto containing the array.ONNXOperators.build
(ONNXContext context, String[] inputs, String output) Builds this node based on the supplied inputs and output.ONNXOperators.build
(ONNXContext context, String[] inputs, String[] outputs) Builds this node based on the supplied inputs and outputs.ONNXOperators.build
(ONNXContext context, String[] inputs, String[] outputs, Map<String, Object> attributeValues) Builds this node based on the supplied inputs and outputs.ONNXOperators.build
(ONNXContext context, String[] inputs, String output, Map<String, Object> attributeValues) Builds this node based on the supplied inputs and output.ONNXOperators.build
(ONNXContext context, String input, String output) Builds this node based on the supplied inputs and output.ONNXOperators.build
(ONNXContext context, String input, String[] outputs) Builds this node based on the supplied input and outputs.ONNXOperators.build
(ONNXContext context, String input, String[] outputs, Map<String, Object> attributeValues) Builds this node based on the supplied input and outputs.ONNXOperators.build
(ONNXContext context, String input, String output, Map<String, Object> attributeValues) Builds this node based on the supplied inputs and output.static OnnxMl.TensorProto
ONNXUtils.doubleTensorBuilder
(ONNXContext context, String name, List<Integer> dims, Consumer<DoubleBuffer> dataPopulator) Generic method to create doubleOnnxMl.TensorProto
instances.static OnnxMl.TensorProto
ONNXUtils.floatTensorBuilder
(ONNXContext context, String name, List<Integer> dims, Consumer<FloatBuffer> dataPopulator) Generic method to create floatOnnxMl.TensorProto
instances.static OnnxMl.TensorProto
ONNXUtils.scalarBuilder
(ONNXContext context, String name, double value) Builds a TensorProto containing the scalar value.static OnnxMl.TensorProto
ONNXUtils.scalarBuilder
(ONNXContext context, String name, float value) Builds a TensorProto containing the scalar value.static OnnxMl.TensorProto
ONNXUtils.scalarBuilder
(ONNXContext context, String name, int value) Builds a TensorProto containing the scalar value.static OnnxMl.TensorProto
ONNXUtils.scalarBuilder
(ONNXContext context, String name, long value) Builds a TensorProto containing the scalar value.