reduce

Documentation

Index

• Variables
• type Ops
  • func Stage(reduceop gocudnn.ReduceTensorOp, dtype gocudnn.DataType, ...) (*Ops, error)
  • func (o *Ops) GetIndiciesSize(handle *cudnn.Handler, x, y *tensor.Volume) (uint, error)
  • func (o *Ops) GetWorkSpaceSize(handle *cudnn.Handler, x, y *tensor.Volume) (uint, error)
  • func (o *Ops) Reduce(handle *cudnn.Handler, indicies *nvidia.Malloced, workspace *nvidia.Malloced, ...) error

Variables

var Flags flagsforop

Flags are the reduce op

Types

type Ops

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

Ops contains the reduce ops information

func Stage

func Stage(reduceop gocudnn.ReduceTensorOp, dtype gocudnn.DataType, nanprop gocudnn.NANProp, reducetensorinds gocudnn.ReduceTensorIndices, indicietype gocudnn.IndiciesType) (*Ops, error)

Stage stages the Reduce Operation

func (*Ops) GetIndiciesSize

func (o *Ops) GetIndiciesSize(handle *cudnn.Handler, x, y *tensor.Volume) (uint, error)

GetIndiciesSize returns the size of indicies

func (*Ops) GetWorkSpaceSize

func (o *Ops) GetWorkSpaceSize(handle *cudnn.Handler, x, y *tensor.Volume) (uint, error)

GetWorkSpaceSize returns the workspace size for the two tensors

func (*Ops) Reduce

func (o *Ops) Reduce(handle *cudnn.Handler, indicies *nvidia.Malloced, workspace *nvidia.Malloced, alpha float64, x *tensor.Volume, beta float64, y *tensor.Volume) error

Reduce performs the reduce operation with input/output being y where y= alpha* Op(x) +beta*y