options

package

v0.2.23-0...-4aaa12c Latest Latest Go to latest Published: May 17, 2020 License: NCSA Imports: 12 Imported by: 39

Details

Valid go.mod file

The Go module system was introduced in Go 1.11 and is the official dependency management solution for Go.
Redistributable license

Redistributable licenses place minimal restrictions on how software can be used, modified, and redistributed.
Tagged version

Modules with tagged versions give importers more predictable builds.
Stable version

When a project reaches major version v1 it is considered stable.
Learn more about best practices

Repository

github.com/rai-project/dlframework

Links

Open Source Insights

Documentation ¶

Constants ¶

This section is empty.

Variables ¶

View Source

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

View Source

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

Functions ¶

This section is empty.

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

Source Files ¶

View all Source files

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