Package org.tribuo.util.onnx

package org.tribuo.util.onnx
Interfaces and utilities for writing ONNX models from Java.

Developed to support Tribuo, but can be used to export other machine learning models from JVM languages.

  • Class
    Description
    ONNXAttribute
    The spec for an attribute, used to produce the attribute proto at construction time.
    ONNXContext
    Context object used to scope and manage the creation of ONNX OnnxMl.GraphProto and OnnxMl.ModelProto instances.
    ONNXInitializer
    A subclass of ONNXRef specialized for OnnxMl.TensorProto.
    ONNXNode
    A subclass of ONNXRef specialized for OnnxMl.NodeProto.
    ONNXOperator
    An interface for ONNX operators.
    ONNXOperators
    ONNX Opset 13, and ONNX-ML version 1.
    ONNXPlaceholder
    A subclass of ONNXRef specialized for OnnxMl.ValueInfoProto.
    ONNXRef<T extends com.google.protobuf.GeneratedMessageV3>
    An abstract reference that represents both a node in an ONNX computation graph and a container for a specific ONNX proto object that denotes that node.
    ONNXUtils
    Helper functions for building ONNX protos.