kernel

Documentation

Index

• func GetModelOps(lib *plugin.Plugin, file []byte) (uint64, error)
• type Model
  • func New(lib *plugin.Plugin, deviceId int, modelCfg, modelBin []byte) *Model
  • func (m *Model) Free()
  • func (m *Model) GetInputLength() int
  • func (m *Model) GetOutputLength() int
  • func (m *Model) Ops() int64
  • func (m *Model) Predict(data []byte) ([]byte, error)
  • func (m *Model) Size() int64

Functions

func GetModelOps

func GetModelOps(lib *plugin.Plugin, file []byte) (uint64, error)

Types

type Model

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

func New

func New(lib *plugin.Plugin, deviceId int, modelCfg, modelBin []byte) *Model

func (*Model) Free

func (m *Model) Free()

func (*Model) GetInputLength

func (m *Model) GetInputLength() int

func (*Model) GetOutputLength

func (m *Model) GetOutputLength() int

func (*Model) Ops

func (m *Model) Ops() int64

func (*Model) Predict

func (m *Model) Predict(data []byte) ([]byte, error)

func (*Model) Size

func (m *Model) Size() int64