paddle

package
v0.0.0-...-d62c078 Latest Latest
Warning

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

 Go to latest Published: Nov 3, 2020 License: Apache-2.0 Imports: 7 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

func ConvertCBooleanToGo 

func ConvertCBooleanToGo(b C.bool) bool

func DecodeTensor 

func DecodeTensor(r *bytes.Reader, shape []int32, t reflect.Type, ptr reflect.Value)

func DeletePredictor 

func DeletePredictor(predictor *Predictor)

func Endian 

func Endian() binary.ByteOrder

func SizeofDataType 

func SizeofDataType(dtype PaddleDType) int32

func TypeOf 

func TypeOf(dtype PaddleDType, shape []int32) reflect.Type

func TypeOfShape 

func TypeOfShape(dtype PaddleDType, shape []int32) reflect.Type

Types

type AnalysisConfig 

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

func NewAnalysisConfig 

func NewAnalysisConfig() *AnalysisConfig

func (*AnalysisConfig) CpuMathLibraryNumThreads 

func (config *AnalysisConfig) CpuMathLibraryNumThreads() int

func (*AnalysisConfig) CudnnEnabled 

func (config *AnalysisConfig) CudnnEnabled() bool

func (*AnalysisConfig) DeletePass 

func (config *AnalysisConfig) DeletePass(pass string)

func (*AnalysisConfig) DisableGlogInfo 

func (config *AnalysisConfig) DisableGlogInfo()

func (*AnalysisConfig) DisableGpu 

func (config *AnalysisConfig) DisableGpu()

func (*AnalysisConfig) EnableCudnn 

func (config *AnalysisConfig) EnableCudnn()

func (*AnalysisConfig) EnableMemoryOptim 

func (config *AnalysisConfig) EnableMemoryOptim()

func (*AnalysisConfig) EnableMkldnn 

func (config *AnalysisConfig) EnableMkldnn()

func (*AnalysisConfig) EnableMkldnnQuantizer 

func (config *AnalysisConfig) EnableMkldnnQuantizer()

func (*AnalysisConfig) EnableProfile 

func (config *AnalysisConfig) EnableProfile()

func (*AnalysisConfig) EnableTensorRtEngine 

func (config *AnalysisConfig) EnableTensorRtEngine(workspace_size int, max_batch_size int, min_subgraph_size int, precision Precision, use_static bool, use_calib_mode bool)

func (*AnalysisConfig) EnableUseGpu 

func (config *AnalysisConfig) EnableUseGpu(memory_pool_init_size_mb int, device_id int)

func (*AnalysisConfig) GpuDeviceId 

func (config *AnalysisConfig) GpuDeviceId() int

func (*AnalysisConfig) IrOptim 

func (config *AnalysisConfig) IrOptim() bool

func (*AnalysisConfig) IsValid 

func (config *AnalysisConfig) IsValid() bool

func (*AnalysisConfig) MemoryOptimEnabled 

func (config *AnalysisConfig) MemoryOptimEnabled() bool

func (*AnalysisConfig) MemoryPoolInitSizeMb 

func (config *AnalysisConfig) MemoryPoolInitSizeMb() int

func (*AnalysisConfig) MkldnnQuantizerEnabled 

func (config *AnalysisConfig) MkldnnQuantizerEnabled() bool

func (*AnalysisConfig) ModelDir 

func (config *AnalysisConfig) ModelDir() string

func (*AnalysisConfig) ParamsFile 

func (config *AnalysisConfig) ParamsFile() string

func (*AnalysisConfig) ProfileEnabled 

func (config *AnalysisConfig) ProfileEnabled() bool

func (*AnalysisConfig) ProgFile 

func (config *AnalysisConfig) ProgFile() string

func (*AnalysisConfig) SetCpuMathLibraryNumThreads 

func (config *AnalysisConfig) SetCpuMathLibraryNumThreads(n int)

func (*AnalysisConfig) SetInValid 

func (config *AnalysisConfig) SetInValid()

func (*AnalysisConfig) SetModel 

func (config *AnalysisConfig) SetModel(model, params string)

func (*AnalysisConfig) SpecifyInputName 

func (config *AnalysisConfig) SpecifyInputName() bool

func (*AnalysisConfig) SwitchIrDebug 

func (config *AnalysisConfig) SwitchIrDebug(x bool)

func (*AnalysisConfig) SwitchIrOptim 

func (config *AnalysisConfig) SwitchIrOptim(x bool)

func (*AnalysisConfig) SwitchSpecifyInputNames 

func (config *AnalysisConfig) SwitchSpecifyInputNames(x bool)

func (*AnalysisConfig) SwitchUseFeedFetchOps 

func (config *AnalysisConfig) SwitchUseFeedFetchOps(x bool)

func (*AnalysisConfig) TensorrtEngineEnabled 

func (config *AnalysisConfig) TensorrtEngineEnabled() bool

func (*AnalysisConfig) UseFeedFetchOpsEnabled 

func (config *AnalysisConfig) UseFeedFetchOpsEnabled() bool

func (*AnalysisConfig) UseGpu 

func (config *AnalysisConfig) UseGpu() bool

type PaddleDType 

type PaddleDType C.PD_DataType
const (
	FLOAT32  PaddleDType = C.PD_FLOAT32
	INT32    PaddleDType = C.PD_INT32
	INT64    PaddleDType = C.PD_INT64
	UINT8    PaddleDType = C.PD_UINT8
	UNKDTYPE PaddleDType = C.PD_UNKDTYPE
)

func ShapeAndTypeOf 

func ShapeAndTypeOf(val reflect.Value) (shape []int32, dt PaddleDType)

type Precision 

type Precision C.Precision
const (
	Precision_FLOAT32 Precision = C.kFloat32
	Precision_INT8    Precision = C.kInt8
	Precision_HALF    Precision = C.kHalf
)

type Predictor 

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

func NewPredictor 

func NewPredictor(config *AnalysisConfig) *Predictor

func (*Predictor) GetInputName 

func (predictor *Predictor) GetInputName(n int) string

func (*Predictor) GetInputNames 

func (predictor *Predictor) GetInputNames() []string

func (*Predictor) GetInputNum 

func (predictor *Predictor) GetInputNum() int

func (*Predictor) GetInputTensors 

func (predictor *Predictor) GetInputTensors() [](*ZeroCopyTensor)

func (*Predictor) GetOutputName 

func (predictor *Predictor) GetOutputName(n int) string

func (*Predictor) GetOutputNames 

func (predictor *Predictor) GetOutputNames() []string

func (*Predictor) GetOutputNum 

func (predictor *Predictor) GetOutputNum() int

func (*Predictor) GetOutputTensors 

func (predictor *Predictor) GetOutputTensors() [](*ZeroCopyTensor)

func (*Predictor) GetZeroCopyOutput 

func (predictor *Predictor) GetZeroCopyOutput(tensor *ZeroCopyTensor)

func (*Predictor) SetZeroCopyInput 

func (predictor *Predictor) SetZeroCopyInput(tensor *ZeroCopyTensor)

func (*Predictor) ZeroCopyRun 

func (predictor *Predictor) ZeroCopyRun()

type ZeroCopyTensor 

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

func NewZeroCopyTensor 

func NewZeroCopyTensor() *ZeroCopyTensor

func (*ZeroCopyTensor) DataType 

func (tensor *ZeroCopyTensor) DataType() PaddleDType

func (*ZeroCopyTensor) Lod 

func (tensor *ZeroCopyTensor) Lod() []uint

func (*ZeroCopyTensor) Name 

func (tensor *ZeroCopyTensor) Name() string

func (*ZeroCopyTensor) Rename 

func (tensor *ZeroCopyTensor) Rename(name string)

func (*ZeroCopyTensor) Reshape 

func (tensor *ZeroCopyTensor) Reshape(shape []int32)

func (*ZeroCopyTensor) SetValue 

func (tensor *ZeroCopyTensor) SetValue(value interface{})

func (*ZeroCopyTensor) Shape 

func (tensor *ZeroCopyTensor) Shape() []int32

func (*ZeroCopyTensor) Value 

func (tensor *ZeroCopyTensor) Value() interface{}

Source Files

View all Source files

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.