options

Documentation

Index

• Variables
• type DeviceType
  • func (d DeviceType) String() string
• type Node
• type Option
  • func BatchSize(n int) Option
  • func Context(ctx context.Context) Option
  • func Device(deviceType DeviceType, id int) Option
  • func DisableFrameworkAutoTuning(disabled bool) Option
  • func FeatureLimit(num int) Option
  • func GPUMetrics(m string) Option
  • func Graph(g []byte) Option
  • func InputNodes(ins []Node) Option
  • func OutputNodes(outs []Node) Option
  • func PredictorOptions(p *dl.PredictionOptions) Option
  • func TraceLevel(tl dl.ExecutionOptions_TraceLevel) Option
  • func Weights(w []byte) Option
  • func WithOptions(opts *Options) Option
• type Options
  • func New(opts ...Option) *Options
  • func (o *Options) Append(opts ...Option) *Options
  • func (o *Options) BatchSize() int
  • func (o *Options) Context() context.Context
  • func (o *Options) Devices() devices
  • func (o *Options) DisableFrameworkAutoTuning() bool
  • func (o *Options) FeatureLimit() int
  • func (o *Options) GPUMetrics() string
  • func (o *Options) Graph() []byte
  • func (o *Options) InputNodes() []Node
  • func (o *Options) OutputIndexes() []int
  • func (o *Options) OutputNodes() []Node
  • func (o *Options) SetBatchSize(n int)
  • func (o *Options) SetContext(ctx context.Context)
  • func (o *Options) SetDevice(deviceType DeviceType, id int)
  • func (o *Options) SetDisableFrameworkAutoTuning(disabled bool)
  • func (o *Options) SetFeatureLimit(n int)
  • func (o *Options) SetGPUMetrics(m string)
  • func (o *Options) SetGraph(g []byte)
  • func (o *Options) SetInputNodes(ins []Node)
  • func (o *Options) SetOutputIndexes(outIdxs []int)
  • func (o *Options) SetOutputNodes(ins []Node)
  • func (o *Options) SetTraceLevel(g tracer.Level)
  • func (o *Options) SetWeights(w []byte)
  • func (o *Options) TraceLevel() tracer.Level
  • func (o *Options) UsesGPU() bool
  • func (o *Options) Weights() []byte

Variables

var (
	DefaultBatchSize    int = 1
	DefaultFeatureLimit int = 10
	DefaultGPUMetrics   string
	DefaultDevice       device
)

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

Types

type DeviceType

type DeviceType int

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

func (DeviceType) String

func (d DeviceType) String() string

type Node

type Node struct {
	Key   string // name
	Shape []int  // Shape of ndarray
	Dtype tensor.Dtype
}

type Option

type Option func(*Options)

func BatchSize

func BatchSize(n int) Option

func Context

func Context(ctx context.Context) Option

func Device

func Device(deviceType DeviceType, id int) Option

func DisableFrameworkAutoTuning

func DisableFrameworkAutoTuning(disabled bool) Option

func FeatureLimit

func FeatureLimit(num int) Option

func GPUMetrics

func GPUMetrics(m string) Option

func Graph

func Graph(g []byte) Option

func InputNodes

func InputNodes(ins []Node) Option

func OutputNodes

func OutputNodes(outs []Node) Option

func PredictorOptions

func PredictorOptions(p *dl.PredictionOptions) Option

func TraceLevel

func TraceLevel(tl dl.ExecutionOptions_TraceLevel) Option

func Weights

func Weights(w []byte) Option

func WithOptions

func WithOptions(opts *Options) Option

type Options

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

func New

func New(opts ...Option) *Options

func (*Options) Append

func (o *Options) Append(opts ...Option) *Options

func (*Options) BatchSize

func (o *Options) BatchSize() int

func (*Options) Context

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

func (*Options) Devices

func (o *Options) Devices() devices

func (*Options) DisableFrameworkAutoTuning

func (o *Options) DisableFrameworkAutoTuning() bool

func (*Options) FeatureLimit

func (o *Options) FeatureLimit() int

func (*Options) GPUMetrics

func (o *Options) GPUMetrics() string

func (*Options) Graph

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

func (*Options) InputNodes

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

func (*Options) OutputIndexes

func (o *Options) OutputIndexes() []int

func (*Options) OutputNodes

func (o *Options) OutputNodes() []Node

func (*Options) SetBatchSize

func (o *Options) SetBatchSize(n int)

func (*Options) SetContext

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

func (*Options) SetDevice

func (o *Options) SetDevice(deviceType DeviceType, id int)

func (*Options) SetDisableFrameworkAutoTuning

func (o *Options) SetDisableFrameworkAutoTuning(disabled bool)

func (*Options) SetFeatureLimit

func (o *Options) SetFeatureLimit(n int)

func (*Options) SetGPUMetrics

func (o *Options) SetGPUMetrics(m string)

func (*Options) SetGraph

func (o *Options) SetGraph(g []byte)

func (*Options) SetInputNodes

func (o *Options) SetInputNodes(ins []Node)

func (*Options) SetOutputIndexes

func (o *Options) SetOutputIndexes(outIdxs []int)

func (*Options) SetOutputNodes

func (o *Options) SetOutputNodes(ins []Node)

func (*Options) SetTraceLevel

func (o *Options) SetTraceLevel(g tracer.Level)

func (*Options) SetWeights

func (o *Options) SetWeights(w []byte)

func (*Options) TraceLevel

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

func (*Options) UsesGPU

func (o *Options) UsesGPU() bool

func (*Options) Weights

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