onnx

package module

v0.0.4 Latest Latest Go to latest Published: Jul 1, 2022 License: MIT Imports: 6 Imported by: 0

Details

Valid go.mod file

The Go module system was introduced in Go 1.11 and is the official dependency management solution for Go.
Redistributable license

Redistributable licenses place minimal restrictions on how software can be used, modified, and redistributed.
Tagged version

Modules with tagged versions give importers more predictable builds.
Stable version

When a project reaches major version v1 it is considered stable.
Learn more about best practices

Repository

github.com/ClearBlade/go-onnx

Links

Open Source Insights

Documentation ¶

Index ¶

type IOInfo
type OnnxRuntime
- func NewOnnxRuntime(model []byte) (*OnnxRuntime, error)
type TensorElementDataType
- func (t TensorElementDataType) Dtype() tensor.Dtype
- func (t TensorElementDataType) String() string
type ValueType
- func (t ValueType) String() string

Constants ¶

This section is empty.

Variables ¶

This section is empty.

Functions ¶

This section is empty.

Types ¶

type IOInfo ¶

type IOInfo struct {
	Name string
	Type ValueType

	// these fields are only valid if Type == ValueTypeTensor
	DataType TensorElementDataType
	Shape    []int
}

type OnnxRuntime ¶

type OnnxRuntime struct {
	Inputs  []IOInfo
	Outputs []IOInfo
	// contains filtered or unexported fields
}

func NewOnnxRuntime ¶

func NewOnnxRuntime(model []byte) (*OnnxRuntime, error)

func (*OnnxRuntime) Cleanup ¶

func (o *OnnxRuntime) Cleanup()

func (*OnnxRuntime) Run ¶

func (o *OnnxRuntime) Run(desiredOutputs []string, inputs map[string]*tensor.Dense) (out map[string]interface{}, err error)

func (*OnnxRuntime) RunSimple ¶

func (o *OnnxRuntime) RunSimple(inputs ...*tensor.Dense) (map[string]interface{}, error)

type TensorElementDataType ¶

type TensorElementDataType int

const (
	TensorElementDataTypeUndefined TensorElementDataType = iota
	TensorElementDataTypeFloat
	TensorElementDataTypeDouble
	TensorElementDataTypeUint8
	TensorElementDataTypeUint16
	TensorElementDataTypeUint32
	TensorElementDataTypeUint64
	TensorElementDataTypeInt8
	TensorElementDataTypeInt16
	TensorElementDataTypeInt32
	TensorElementDataTypeInt64
	TensorElementDataTypeBool
	TensorElementDataTypeString
	TensorElementDataTypeUnsupported
	TensorElementDataTypeInvalid
)

func (TensorElementDataType) Dtype ¶

func (t TensorElementDataType) Dtype() tensor.Dtype

func (TensorElementDataType) String ¶

func (t TensorElementDataType) String() string

type ValueType ¶

type ValueType int

const (
	ValueTypeUnknown ValueType = iota
	ValueTypeTensor
	ValueTypeSequence
	ValueTypeMap
	ValueTypeOpaque
	ValueTypeSparseTensor
	ValueTypeOptional
)

func (ValueType) String ¶

func (t ValueType) String() string

Source Files ¶

View all Source files

?	: This menu
/	: Search site
f or F	: Jump to
y or Y	: Canonical URL