tensorrt

package module

v0.3.20 Latest Latest Go to latest Published: Jul 16, 2018 License: NCSA Imports: 12 Imported by: 0

Details

Rendered for

This section is empty.

View Source

var (
	Version   = "0.3.20"
	BuildDate = "undefined"
	GitCommit = "undefined"
)

This section is empty.

type Prediction struct {
	Index       int     `json:"index"`
	Probability float32 `json:"probability"`
}

type Predictions []Prediction

func (p Predictions) Len() int

Len is the number of elements in the collection.

func (p Predictions) Less(i, j int) bool

Less reports whether the element with index i should sort before the element with index j.

func (p Predictions) Sort()

func (p Predictions) Swap(i, j int)

Swap swaps the elements with indexes i and j.

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

func New(opts0 ...options.Option) (*Predictor, error)

func (p Predictor) Close()

func (p *Predictor) DisableProfiling() error

func (p *Predictor) EndProfiling() error

func (p *Predictor) Predict(inputLayerName0 string, outputLayerName0 string, input []float32, shape []uint32) (Predictions, error)

func (p *Predictor) ReadProfile() (string, error)

func (p *Predictor) StartProfiling(name, metadata string) error