fakemodel

Documentation

Index

• Variables
• type DoubleBuffer
  • func NewDoubleBuffer(dtype kb.DataType, count int) DoubleBuffer
• type FakeModel
  • func New(sizes []int, dtype kb.DataType, fuse bool) *FakeModel
  • func (m *FakeModel) Info() string
  • func (m *FakeModel) MaxSize() int
  • func (m *FakeModel) MinSize() int
  • func (m *FakeModel) ShowInfo()
  • func (m *FakeModel) Size() int

Variables

var Models = map[string][]int{
	"resnet50-imagenet": resnet50Imagenet,
	"vgg16-imagenet":    vgg16Imagenet,
	"slp-mnist":         slpMNIST,
	"bert":              bert,
}

var Names = func(m map[string][]int) []string {
	var ks []string
	for k := range m {
		ks = append(ks, k)
	}
	sort.Strings(ks)
	return ks
}(Models)

Types

type DoubleBuffer

type DoubleBuffer struct {
	SendBuf *kb.Vector
	RecvBuf *kb.Vector
}

func NewDoubleBuffer

func NewDoubleBuffer(dtype kb.DataType, count int) DoubleBuffer

type FakeModel

type FakeModel struct {
	Names   []string
	Buffers map[string]DoubleBuffer
}

func New

func New(sizes []int, dtype kb.DataType, fuse bool) *FakeModel

func (*FakeModel) Info

func (m *FakeModel) Info() string

func (*FakeModel) MaxSize

func (m *FakeModel) MaxSize() int

func (*FakeModel) MinSize

func (m *FakeModel) MinSize() int

func (*FakeModel) ShowInfo

func (m *FakeModel) ShowInfo()

func (*FakeModel) Size

func (m *FakeModel) Size() int