onnx

type Attr struct {
	Name   string
	Type   AttrType
	F      float32
	I      int64
	S      string
	T      *Tensor
	Floats []float32
	Ints   []int64
}

type AttrType int32

type DataType int32

type Graph struct {
	Name         string
	Nodes        []*Node
	Initializers map[string]*Tensor
	Inputs       []ValueInfo
	Outputs      []ValueInfo
}

type Limits struct {
	// ModelBytes bounds the encoded model.
	ModelBytes int
	// Nodes bounds the graph's nodes.
	Nodes int
	// Dims bounds a tensor's rank.
	Dims int
	// TensorElems bounds one tensor's elements (initializers, inputs and
	// every intermediate result).
	TensorElems int64
	// TotalElems bounds the sum of the initializers' elements.
	TotalElems int64
	// MACs bounds the multiply-accumulates of one inference.
	MACs int64
	// Strings bounds a name or attribute string, in bytes.
	Strings int
}

type Model struct {
	IRVersion int64
	// Opsets maps an operator-set domain ("" is the default ai.onnx) to
	// its version.
	Opsets   map[string]int64
	Producer string
	Graph    *Graph
}

type Node struct {
	Name, Op, Domain string
	Inputs, Outputs  []string
	Attrs            map[string]*Attr
}

type Plan struct {
	Opset  int64
	Steps  []Step
	Shapes map[string][]int64
	Consts map[string]*Tensor
}

type Session struct {
	// contains filtered or unexported fields
}

type Step struct {
	Node      *Node
	InShapes  [][]int64 // nil for an omitted optional input
	OutShapes [][]int64
	// Const says the outputs do not depend on the graph inputs; they are
	// in Plan.Consts.
	Const bool
}

type Tensor struct {
	Name string
	Dims []int64
	Type DataType
	F    []float32
	I    []int64
}

type UnsupportedError struct {
	Ops map[string]int // "domain:Op" or "Op" -> nodes
}

type ValueInfo struct {
	Name string
	Type DataType
	Dims []int64
}

type Window struct {
	In, Kernel, Stride, Dil, PadB, PadE, Out []int64
}