Documentation
¶
Overview ¶
Package cudnn takes the descriptors from gocudnn which is from cudnn and seperates them into seperate packages. In the hopes to eventually move away from all those pesky flags. Some flags will have to exist still, but where I can get rid of them I will. Like DataType and NanProp and TensorFormat, and even algorithm for convolution.
Index ¶
- func ToImagesColorFloat32(data []float32, dims []int32, frmt gocudnn.TensorFormat) ([][]image.Image, error)
- type Handler
- func (h *Handler) Cudnn() *gocudnn.Handle
- func (h *Handler) DeviceSync() error
- func (h *Handler) FindVol(outputdims []int32) int32
- func (h *Handler) GetCuRNG() *curand.Generator
- func (h *Handler) MakeNotUnified()
- func (h *Handler) SetDevice() error
- func (h *Handler) SetStream(stream gocu.Streamer) error
- func (h *Handler) Stream() gocu.Streamer
- func (h *Handler) Sync() error
- func (h *Handler) SyncContext() error
- func (h *Handler) Unified() bool
- func (h *Handler) XHandle() *xtra.Handle
- type TrainMode
- type TrainModeFlag
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
func ToImagesColorFloat32 ¶
func ToImagesColorFloat32(data []float32, dims []int32, frmt gocudnn.TensorFormat) ([][]image.Image, error)
ToImagesColorFloat32 changes data passed to a bunch of images
Types ¶
type Handler ¶
Handler contains the handles used in gocudnn and also the xtra kernals.
func CreateHandler ¶
CreateHandler creates a the handlers The handler is used in managing memory for all the packages that use cudnn.Handler. This function will raise a flag that will tell the program to use unified memory management. If that is not wanted call MakeNotUnified immediately to turn this off.
func (*Handler) FindVol ¶
FindVol will find the max vol for tensor. This is going to hold two functions pmax can be either the previous maxvol, or it could be
func (*Handler) MakeNotUnified ¶
func (h *Handler) MakeNotUnified()
MakeNotUnified sets unified flag for device the handler is using to false.
func (*Handler) SyncContext ¶
SyncContext will sync the contexts
type TrainMode ¶
type TrainMode xtra.TrainingMode
TrainMode is a flag to set the TrainingMode
func (TrainMode) Cu ¶
func (t TrainMode) Cu() xtra.TrainingMode
Cu is used to pass gocudnn.TrainMode flags for the functions inside the different cudnn packages
type TrainModeFlag ¶
type TrainModeFlag struct {
// contains filtered or unexported fields
}
TrainModeFlag is used to pass TrainingMode flags through methods.
func (TrainModeFlag) AdaDelta ¶
func (t TrainModeFlag) AdaDelta() TrainMode
AdaDelta is used to set the TrainingMode to AdaDelta
func (TrainModeFlag) AdaGrad ¶
func (t TrainModeFlag) AdaGrad() TrainMode
AdaGrad is used to set the TrainingMode to AdaGrad
func (TrainModeFlag) Adam ¶
func (t TrainModeFlag) Adam() TrainMode
Adam is used to set the TrainingMode to Adam
Source Files
Directories
¶
| Path | Synopsis |
|---|---|
|
Package softmax uses the softmax functions from gocudnn which is from cudnn.
|
Package softmax uses the softmax functions from gocudnn which is from cudnn. |
|
Package tensor is used to make tensors by using gocudnn.
|
Package tensor is used to make tensors by using gocudnn. |