consts

Documentation

Index

• type DeviceType
  • func (t DeviceType) String() string
• type Reduction
• type ScalarType
  • func (t ScalarType) String() string

Types

type DeviceType

type DeviceType byte

const (
	KCPU DeviceType = iota
	KCUDA
	KMKLDNN
	KOPENGL
	KOPENCL
	KIDEEP
	KHIP
	KFPGA
	KORT
	KXLA
	KVulkan
	KMetal
	KXPU
	KMPS
	KMeta
	KHPU
	KVE
	KLazy
	KIPU
	KMTIA
)

func (DeviceType) String

func (t DeviceType) String() string

type Reduction

type Reduction byte

const (
	ReductionNone Reduction = iota
	ReductionMean
	ReductionSum
	ReductionEnd
)

type ScalarType

type ScalarType byte

const (
	KUint8 ScalarType = iota
	KInt8
	KInt16
	KInt32
	KInt64
	KHalf
	KFloat
	KDouble
	KComplexHalf
	KComplexFloat
	KComplexDouble
	KBool
	KQInt8
	KQUInt8
	KQInt32
	KBFloat16
)

func (ScalarType) String

func (t ScalarType) String() string