Uses of Class
org.tribuo.util.onnx.ONNXContext
Packages that use 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
Methods in org.tribuo with parameters of type ONNXContextModifier 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
Methods in org.tribuo.math.onnx with parameters of type ONNXContextModifier and TypeMethodDescriptionstatic ONNXInitializerONNXMathUtils.floatMatrix(ONNXContext context, String name, Matrix matrix, boolean transpose) Builds aONNXInitializercontaining theMatrix.static ONNXInitializerONNXMathUtils.floatVector(ONNXContext context, String name, SGDVector vector) Builds aONNXInitializercontaining theSGDVector. -
Uses of ONNXContext in org.tribuo.util.onnx
Fields in org.tribuo.util.onnx declared as ONNXContextMethods in org.tribuo.util.onnx that return ONNXContextMethods in org.tribuo.util.onnx with parameters of type ONNXContextModifier and TypeMethodDescriptionstatic ai.onnx.proto.OnnxMl.TensorProtoONNXUtils.arrayBuilder(ONNXContext context, String name, double[] parameters) Builds a TensorProto containing the array.static ai.onnx.proto.OnnxMl.TensorProtoONNXUtils.arrayBuilder(ONNXContext context, String name, double[] parameters, boolean downcast) Builds a TensorProto containing the array.static ai.onnx.proto.OnnxMl.TensorProtoONNXUtils.arrayBuilder(ONNXContext context, String name, float[] parameters) Builds a TensorProto containing the array.static ai.onnx.proto.OnnxMl.TensorProtoONNXUtils.arrayBuilder(ONNXContext context, String name, int[] parameters) Builds a TensorProto containing the array.static ai.onnx.proto.OnnxMl.TensorProtoONNXUtils.arrayBuilder(ONNXContext context, String name, long[] parameters) Builds a TensorProto containing the array.default ai.onnx.proto.OnnxMl.NodeProtoONNXOperator.build(ONNXContext context, String[] inputs, String output) Builds this node based on the supplied inputs and output.default ai.onnx.proto.OnnxMl.NodeProtoONNXOperator.build(ONNXContext context, String[] inputs, String[] outputs) Builds this node based on the supplied inputs and outputs.default ai.onnx.proto.OnnxMl.NodeProtoONNXOperator.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.NodeProtoONNXOperator.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.NodeProtoONNXOperator.build(ONNXContext context, String input, String output) Builds this node based on the supplied inputs and output.default ai.onnx.proto.OnnxMl.NodeProtoONNXOperator.build(ONNXContext context, String input, String[] outputs) Builds this node based on the supplied input and outputs.default ai.onnx.proto.OnnxMl.NodeProtoONNXOperator.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.NodeProtoONNXOperator.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.TensorProtoONNXUtils.doubleTensorBuilder(ONNXContext context, String name, List<Integer> dims, Consumer<DoubleBuffer> dataPopulator) Generic method to create doubleOnnxMl.TensorProtoinstances.static ai.onnx.proto.OnnxMl.TensorProtoONNXUtils.floatTensorBuilder(ONNXContext context, String name, List<Integer> dims, Consumer<FloatBuffer> dataPopulator) Generic method to create floatOnnxMl.TensorProtoinstances.static ai.onnx.proto.OnnxMl.TensorProtoONNXUtils.scalarBuilder(ONNXContext context, String name, double value) Builds a TensorProto containing the scalar value.static ai.onnx.proto.OnnxMl.TensorProtoONNXUtils.scalarBuilder(ONNXContext context, String name, float value) Builds a TensorProto containing the scalar value.static ai.onnx.proto.OnnxMl.TensorProtoONNXUtils.scalarBuilder(ONNXContext context, String name, int value) Builds a TensorProto containing the scalar value.static ai.onnx.proto.OnnxMl.TensorProtoONNXUtils.scalarBuilder(ONNXContext context, String name, long value) Builds a TensorProto containing the scalar value.