tensorrt

package
v0.0.0-...-2b09442 Latest Latest
Warning

This package is not in the latest version of its module.

Go to latest
Published: Nov 19, 2020 License: Apache-2.0 Imports: 4 Imported by: 1

Documentation

Index

Constants

This section is empty.

Variables

View Source
var (
	ErrorInit    = errors.New("init error")
	ErrorEnqueue = errors.New("enqueue error")
	ErrorParams  = errors.New("params error")
)

TODO: more errors

Functions

func Destory

func Destory(engine Engine)

func Enqueue

func Enqueue(engine Engine, batchSize int, bindings []unsafe.Pointer, stream cuda.Stream, inputComsumed *cuda.Event) error

Types

type Engine

type Engine unsafe.Pointer

func Create

func Create(logger Logger, blob []byte, pluginFactory unsafe.Pointer) (Engine, error)

TODO: pluginFactory need wrap

type Logger

type Logger unsafe.Pointer
var GLogger Logger = Logger(C.gLogger)

Jump to

Keyboard shortcuts

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