Uses of Class
org.tribuo.util.onnx.ONNXContext
Packages that use 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
Methods in org.tribuo with parameters of type ONNXContextModifier 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
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 OnnxMl.TensorProtoONNXUtils.arrayBuilder(ONNXContext context, String name, double[] parameters) Builds a TensorProto containing the array.static OnnxMl.TensorProtoONNXUtils.arrayBuilder(ONNXContext context, String name, double[] parameters, boolean downcast) Builds a TensorProto containing the array.static OnnxMl.TensorProtoONNXUtils.arrayBuilder(ONNXContext context, String name, float[] parameters) Builds a TensorProto containing the array.static OnnxMl.TensorProtoONNXUtils.arrayBuilder(ONNXContext context, String name, int[] parameters) Builds a TensorProto containing the array.static OnnxMl.TensorProtoONNXUtils.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.TensorProtoONNXUtils.doubleTensorBuilder(ONNXContext context, String name, List<Integer> dims, Consumer<DoubleBuffer> dataPopulator) Generic method to create doubleOnnxMl.TensorProtoinstances.static OnnxMl.TensorProtoONNXUtils.floatTensorBuilder(ONNXContext context, String name, List<Integer> dims, Consumer<FloatBuffer> dataPopulator) Generic method to create floatOnnxMl.TensorProtoinstances.static OnnxMl.TensorProtoONNXUtils.scalarBuilder(ONNXContext context, String name, double value) Builds a TensorProto containing the scalar value.static OnnxMl.TensorProtoONNXUtils.scalarBuilder(ONNXContext context, String name, float value) Builds a TensorProto containing the scalar value.static OnnxMl.TensorProtoONNXUtils.scalarBuilder(ONNXContext context, String name, int value) Builds a TensorProto containing the scalar value.static OnnxMl.TensorProtoONNXUtils.scalarBuilder(ONNXContext context, String name, long value) Builds a TensorProto containing the scalar value.