Package onnx

Class OnnxMl

java.lang.Object

onnx.OnnxMl

public final class OnnxMl
extends Object

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	OnnxMl.AttributeProto	Attributes A named attribute containing either singular float, integer, string, graph, and tensor values, or repeated float, integer, string, graph, and tensor values.
static interface	OnnxMl.AttributeProtoOrBuilder
static class	OnnxMl.FunctionProto	Protobuf type onnx.FunctionProto
static interface	OnnxMl.FunctionProtoOrBuilder
static class	OnnxMl.GraphProto	Graphs A graph defines the computational logic of a model and is comprised of a parameterized list of nodes that form a directed acyclic graph based on their inputs and outputs.
static interface	OnnxMl.GraphProtoOrBuilder
static class	OnnxMl.ModelProto	Models ModelProto is a top-level file/container format for bundling a ML model and associating its computation graph with metadata.
static interface	OnnxMl.ModelProtoOrBuilder
static class	OnnxMl.NodeProto	Nodes Computation graphs are made up of a DAG of nodes, which represent what is commonly called a "layer" or "pipeline stage" in machine learning frameworks.
static interface	OnnxMl.NodeProtoOrBuilder
static class	OnnxMl.OperatorSetIdProto	Operator Sets OperatorSets are uniquely identified by a (domain, opset_version) pair.
static interface	OnnxMl.OperatorSetIdProtoOrBuilder
static class	OnnxMl.OperatorStatus	Operator/function status.
static class	OnnxMl.SparseTensorProto	A serialized sparse-tensor value
static interface	OnnxMl.SparseTensorProtoOrBuilder
static class	OnnxMl.StringStringEntryProto	StringStringEntryProto follows the pattern for cross-proto-version maps.
static interface	OnnxMl.StringStringEntryProtoOrBuilder
static class	OnnxMl.TensorAnnotation	Protobuf type onnx.TensorAnnotation
static interface	OnnxMl.TensorAnnotationOrBuilder
static class	OnnxMl.TensorProto	Tensors A serialized tensor value.
static interface	OnnxMl.TensorProtoOrBuilder
static class	OnnxMl.TensorShapeProto	Defines a tensor shape.
static interface	OnnxMl.TensorShapeProtoOrBuilder
static class	OnnxMl.TrainingInfoProto	Training information TrainingInfoProto stores information for training a model.
static interface	OnnxMl.TrainingInfoProtoOrBuilder
static class	OnnxMl.TypeProto	Types The standard ONNX data types.
static interface	OnnxMl.TypeProtoOrBuilder
static class	OnnxMl.ValueInfoProto	Defines information on value, including the name, the type, and the shape of the value.
static interface	OnnxMl.ValueInfoProtoOrBuilder
static class	OnnxMl.Version	Versioning ONNX versioning is specified in docs/IR.md and elaborated on in docs/Versioning.md To be compatible with both proto2 and proto3, we will use a version number that is not defined by the default value but an explicit enum number.

Method Summary

Modifier and Type	Method	Description
static org.nd4j.shade.protobuf.Descriptors.FileDescriptor	getDescriptor()
static void	registerAllExtensions(org.nd4j.shade.protobuf.ExtensionRegistry registry)
static void	registerAllExtensions(org.nd4j.shade.protobuf.ExtensionRegistryLite registry)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

registerAllExtensions

public static void registerAllExtensions(org.nd4j.shade.protobuf.ExtensionRegistryLite registry)

registerAllExtensions

public static void registerAllExtensions(org.nd4j.shade.protobuf.ExtensionRegistry registry)

getDescriptor

public static org.nd4j.shade.protobuf.Descriptors.FileDescriptor getDescriptor()