rknnlite

type AttrMaxDimensions int

type CoreMask int

type ErrorCodes int

type IONumber struct {
	NumberInput  uint32
	NumberOutput uint32
}

type Input struct {
	// Index is the input index
	Index uint32
	// Buf is the gocv Mat input
	Buf unsafe.Pointer
	// Size is the number of bytes of Buf
	Size uint32
	// Passthrough defines the mode, if True the buf data is passed directly to
	// the input node of the rknn model without any conversion.  If False the
	// buf data is converted into an input consistent with the model according
	// to the following type and fmt
	PassThrough bool
	// Type is the data type of Buf. This is a required parameter if Passthrough
	// is False
	Type TensorType
	// Fmt is the data format of Buf.  This is a required parameter if Passthrough
	// is False
	Fmt TensorFormat
}

type Output struct {
	WantFloat  uint8     // want transfer output data to float
	IsPrealloc uint8     // whether buf is pre-allocated
	Index      uint32    // the output index
	Buf        []float32 // the output buf
	Size       uint32    // the size of output buf
}

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

type Probability struct {
	LabelIndex  int32
	Probability float32
}

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

type SDKVersion struct {
	DriverVersion string
	APIVersion    string
}

type TensorAttr struct {
	Index          uint32
	NDims          uint32
	Dims           [AttrMaxDimension]uint32
	Name           string
	NElems         uint32
	Size           uint32
	Fmt            TensorFormat
	Type           TensorType
	QntType        TensorQntType
	FL             int8
	ZP             int32
	Scale          float32
	WStride        uint32
	SizeWithStride uint32
	PassThrough    bool
	HStride        uint32
}

type TensorFormat int

type TensorQntType int

type TensorType int