Uses of Class
org.tribuo.util.onnx.ONNXContext
Package
Description
Provides the core interfaces and classes for using Tribuo.
Utilities for converting Tribuo math objects into ONNX representations.
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>>
ai.onnx.proto.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 ai.onnx.proto.OnnxMl.TensorProto
ONNXUtils.arrayBuilder
(ONNXContext context, String name, double[] parameters) Builds a TensorProto containing the array.static ai.onnx.proto.OnnxMl.TensorProto
ONNXUtils.arrayBuilder
(ONNXContext context, String name, double[] parameters, boolean downcast) Builds a TensorProto containing the array.static ai.onnx.proto.OnnxMl.TensorProto
ONNXUtils.arrayBuilder
(ONNXContext context, String name, float[] parameters) Builds a TensorProto containing the array.static ai.onnx.proto.OnnxMl.TensorProto
ONNXUtils.arrayBuilder
(ONNXContext context, String name, int[] parameters) Builds a TensorProto containing the array.static ai.onnx.proto.OnnxMl.TensorProto
ONNXUtils.arrayBuilder
(ONNXContext context, String name, long[] parameters) Builds a TensorProto containing the array.default ai.onnx.proto.OnnxMl.NodeProto
ONNXOperator.build
(ONNXContext context, String[] inputs, String output) Builds this node based on the supplied inputs and output.default ai.onnx.proto.OnnxMl.NodeProto
ONNXOperator.build
(ONNXContext context, String[] inputs, String[] outputs) Builds this node based on the supplied inputs and outputs.default ai.onnx.proto.OnnxMl.NodeProto
ONNXOperator.build
(ONNXContext context, String[] inputs, String[] outputs, Map<String, Object> attributeValues) Builds this node based on the supplied inputs and outputs.default ai.onnx.proto.OnnxMl.NodeProto
ONNXOperator.build
(ONNXContext context, String[] inputs, String output, Map<String, Object> attributeValues) Builds this node based on the supplied inputs and output.default ai.onnx.proto.OnnxMl.NodeProto
ONNXOperator.build
(ONNXContext context, String input, String output) Builds this node based on the supplied inputs and output.default ai.onnx.proto.OnnxMl.NodeProto
ONNXOperator.build
(ONNXContext context, String input, String[] outputs) Builds this node based on the supplied input and outputs.default ai.onnx.proto.OnnxMl.NodeProto
ONNXOperator.build
(ONNXContext context, String input, String[] outputs, Map<String, Object> attributeValues) Builds this node based on the supplied input and outputs.default ai.onnx.proto.OnnxMl.NodeProto
ONNXOperator.build
(ONNXContext context, String input, String output, Map<String, Object> attributeValues) Builds this node based on the supplied inputs and output.static ai.onnx.proto.OnnxMl.TensorProto
ONNXUtils.doubleTensorBuilder
(ONNXContext context, String name, List<Integer> dims, Consumer<DoubleBuffer> dataPopulator) Generic method to create doubleOnnxMl.TensorProto
instances.static ai.onnx.proto.OnnxMl.TensorProto
ONNXUtils.floatTensorBuilder
(ONNXContext context, String name, List<Integer> dims, Consumer<FloatBuffer> dataPopulator) Generic method to create floatOnnxMl.TensorProto
instances.static ai.onnx.proto.OnnxMl.TensorProto
ONNXUtils.scalarBuilder
(ONNXContext context, String name, double value) Builds a TensorProto containing the scalar value.static ai.onnx.proto.OnnxMl.TensorProto
ONNXUtils.scalarBuilder
(ONNXContext context, String name, float value) Builds a TensorProto containing the scalar value.static ai.onnx.proto.OnnxMl.TensorProto
ONNXUtils.scalarBuilder
(ONNXContext context, String name, int value) Builds a TensorProto containing the scalar value.static ai.onnx.proto.OnnxMl.TensorProto
ONNXUtils.scalarBuilder
(ONNXContext context, String name, long value) Builds a TensorProto containing the scalar value.