coremlcompiler

type Argument struct {
	Bindings []Binding // field 1
}

type ArrayDataType int32

type ArrayFeatureType struct {
	Shape    []int64
	DataType ArrayDataType
}

type Binding struct {
	Name  string // field 1 (oneof)
	Value *Value // field 2 (oneof)
}

type BlobFileValue struct {
	FileName string // field 1
	Offset   uint64 // field 2
}

type Block struct {
	Inputs     []NamedValueType // field 1
	Outputs    []string         // field 2
	Operations []*Operation     // field 3
}

type DataType int32

type Dimension struct {
	Constant uint64 // from ConstantDimension.size (field 1.1)
	Unknown  bool   // true if UnknownDimension
}

type FeatureDescription struct {
	Name string       // field 1
	Type *FeatureType // field 3
}

type FeatureType struct {
	MultiArrayType *ArrayFeatureType
	IsOptional     bool
}

type Function struct {
	Inputs               []NamedValueType  // field 1
	OpSet                string            // field 2
	BlockSpecializations map[string]*Block // field 3
	Attributes           map[string]*Value // field 4
}

type ImmediateValue struct {
	// Exactly one of these is set.
	Tensor *TensorValue // field 1
}

type Model struct {
	SpecVersion int32            // field 1
	Description ModelDescription // field 2
	MLProgram   *Program         // field 502 (oneof)
	// contains filtered or unexported fields
}

type ModelDescription struct {
	Inputs  []FeatureDescription // field 1
	Outputs []FeatureDescription // field 10
	States  []FeatureDescription // field 13
}

type NamedValueType struct {
	Name string     // field 1
	Type *ValueType // field 2
}

type Operation struct {
	Type       string               // field 1
	Inputs     map[string]*Argument // field 2
	Outputs    []NamedValueType     // field 3
	Blocks     []*Block             // field 4
	Attributes map[string]*Value    // field 5
}

type Program struct {
	Version    int64                // field 1
	Functions  map[string]*Function // field 2
	Attributes map[string]*Value    // field 4
}

type StateType struct {
	WrappedType *ValueType // field 1
}

type TensorType struct {
	DataType   DataType    // field 1
	Rank       int64       // field 2
	Dimensions []Dimension // field 3
}

type TensorValue struct {
	// Exactly one of these is set.
	Floats  []float32 // field 1
	Ints    []int32   // field 2
	Bools   []bool    // field 3
	Strings []string  // field 4
	Longs   []int64   // field 5
	Doubles []float64 // field 6
	Bytes   []byte    // field 7
}

type Value struct {
	Type *ValueType // field 2

	// Exactly one of these is set.
	Immediate *ImmediateValue // field 3
	BlobFile  *BlobFileValue  // field 5
}

type ValueType struct {
	// Exactly one of these is set.
	TensorType *TensorType // field 1
	StateType  *StateType  // field 5

}