sut

Documentation

Index

• type SUT
  • func NewSUT(ctx context.Context, backendName string, modelPath string, useGPU bool, ...) (*SUT, error)
  • func (s *SUT) Close()
  • func (s *SUT) GetModelManifest() (dl.ModelManifest, error)
  • func (s *SUT) GetPreprocessMethod() (string, error)
  • func (s *SUT) ProcessQuery(ctx context.Context, data map[int]interface{}, sampleList []int) string

Types

type SUT

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

func NewSUT

func NewSUT(ctx context.Context, backendName string, modelPath string,
	useGPU bool, GPUID int, traceLevel string, batchSize int) (*SUT, error)

NewSUT ...

func (*SUT) Close

func (s *SUT) Close()

func (*SUT) GetModelManifest

func (s *SUT) GetModelManifest() (dl.ModelManifest, error)

func (*SUT) GetPreprocessMethod

func (s *SUT) GetPreprocessMethod() (string, error)

func (*SUT) ProcessQuery

func (s *SUT) ProcessQuery(ctx context.Context, data map[int]interface{}, sampleList []int) string