swapper

Documentation

Index

• type Swapper
  • func CreateSwapper(handle *cudnn.Handler) (*Swapper, error)
  • func (s *Swapper) EveryOther(handle *cudnn.Handler, A, B *tensor.Volume, start, stride int32) error
  • func (s *Swapper) UpperLower(handle *cudnn.Handler, A, B *tensor.Volume, Aupper, Bupper, inverse bool) error

Types

type Swapper

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

Swapper holds kernels that do swap functions

func CreateSwapper

func CreateSwapper(handle *cudnn.Handler) (*Swapper, error)

CreateSwapper returns a Swapper

func (*Swapper) EveryOther

func (s *Swapper) EveryOther(handle *cudnn.Handler, A, B *tensor.Volume, start, stride int32) error

EveryOther swaps either the even or odd of every other batch between two tensors supports float and int8

func (*Swapper) UpperLower

func (s *Swapper) UpperLower(handle *cudnn.Handler, A, B *tensor.Volume, Aupper, Bupper, inverse bool) error

UpperLower swaps either the upper or lower half of the batches between to tensors supports float and int8