options

package

v0.2.20 Latest Latest Go to latest Published: Jun 20, 2018 License: NCSA Imports: 11 Imported by: 11

Details

This section is empty.

View Source

var (
	DefaultBatchSize uint32 = 1
	DefaultDevice    device
)

View Source

var (
	Config = &optionsConfig{
		done: make(chan struct{}),
	}
)

This section is empty.

type DeviceType int

const (
	CPU_DEVICE  DeviceType = iota + 1 // cpu device type
	CUDA_DEVICE DeviceType = 2        // cuda device type
)

func (d DeviceType) String() string

type Option func(*Options)

func BatchSize(n uint32) Option

func Context(c context.Context) Option

func Device(deviceType DeviceType, id int) Option

func Graph(sym []byte) Option

func InputNode(key string, shape []uint32) Option

func OutputNode(output string) Option

func PredictorOptions(p *dl.PredictionOptions) Option

func Symbol(sym []byte) Option

func TraceLevel(tl dl.ExecutionOptions_TraceLevel) Option

func Weights(w []byte) Option

func WithOptions(opts *Options) Option

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

func New(opts ...Option) *Options

func (o *Options) BatchSize() uint32

func (o *Options) Context() context.Context

func (o *Options) Devices() devices

func (o *Options) Graph() []byte

func (o *Options) InputNodes() []inputNode

func (o *Options) OutputNode() string

func (o *Options) Symbol() []byte

func (o *Options) TraceLevel() tracer.Level

func (o *Options) UsesGPU() bool

func (o *Options) Weights() []byte