cudnn

package

Go to main page

Versions in this module

v0

v0.10.0-working

Aug 4, 2021

v0.9.4

Aug 2, 2021

Changes in this version

type Activation

+ func (a *Activation) C() C.cudnnActivationDescriptor_t

type ActivationMode

+ const Identity

+ type Algorithm interface

+ type AlgorithmDescriptor struct

+ func NewAlgorithmDescriptor(algorithm Algorithm) (retVal *AlgorithmDescriptor, err error)

+ func (a *AlgorithmDescriptor) C() C.cudnnAlgorithmDescriptor_t

+ type AlgorithmPerformance struct

+ func NewAlgorithmPerformance(algoDesc *AlgorithmDescriptor, n int, status Status, time float32, ...) (retVal *AlgorithmPerformance, err error)

+ func (a *AlgorithmPerformance) AlgoDesc() *AlgorithmDescriptor

+ func (a *AlgorithmPerformance) C() C.cudnnAlgorithmPerformance_t

+ func (a *AlgorithmPerformance) Memory() uintptr

+ func (a *AlgorithmPerformance) N() int

+ func (a *AlgorithmPerformance) Status() Status

+ func (a *AlgorithmPerformance) Time() float32

+ type Attention struct

+ func NewAttention(attnMode uint, nHeads int, smScaler float64, dataType DataType, ...) (retVal *Attention, err error)

+ func (a *Attention) AttnDropoutDesc() *Dropout

+ func (a *Attention) AttnMode() uint

+ func (a *Attention) C() C.cudnnAttnDescriptor_t

+ func (a *Attention) ComputePrec() DataType

+ func (a *Attention) DataType() DataType

+ func (a *Attention) KProjSize() int

+ func (a *Attention) KSize() int

+ func (a *Attention) KvMaxSeqLength() int

+ func (a *Attention) MathType() MathType

+ func (a *Attention) MaxBatchSize() int

+ func (a *Attention) MaxBeamSize() int

+ func (a *Attention) NHeads() int

+ func (a *Attention) OProjSize() int

+ func (a *Attention) PostDropoutDesc() *Dropout

+ func (a *Attention) QProjSize() int

+ func (a *Attention) QSize() int

+ func (a *Attention) QoMaxSeqLength() int

+ func (a *Attention) SmScaler() float64

+ func (a *Attention) VProjSize() int

+ func (a *Attention) VSize() int

+ type Backend struct

+ func NewBackend(attributeName BackendAttributeName, attributeType BackendAttributeType, ...) (retVal *Backend, err error)

+ func (b *Backend) ArrayOfElements() Memory

+ func (b *Backend) AttributeName() BackendAttributeName

+ func (b *Backend) AttributeType() BackendAttributeType

+ func (b *Backend) C() C.cudnnBackendDescriptor_t

+ func (b *Backend) ElementCount() int64

+ func (b *Backend) Type() BackendDescriptorType

+ type BackendAttributeName int

+ const BEAttrNameConvolutionCompType

+ const BEAttrNameConvolutionConvMode

+ const BEAttrNameConvolutionDilations

+ const BEAttrNameConvolutionFilterStrides

+ const BEAttrNameConvolutionPostPaddings

+ const BEAttrNameConvolutionPrePaddings

+ const BEAttrNameConvolutionSpatialDims

+ const BEAttrNameEngineGlobalIndex

+ const BEAttrNameEngineKnobInfo

+ const BEAttrNameEngineLayoutInfo

+ const BEAttrNameEngineNumericalNote

+ const BEAttrNameEngineOperationGraph

+ const BEAttrNameEnginecfgEngine

+ const BEAttrNameEnginecfgIntermediateInfo

+ const BEAttrNameEnginecfgKnobChoices

+ const BEAttrNameEngineheurMode

+ const BEAttrNameEngineheurOperationGraph

+ const BEAttrNameEngineheurResults

+ const BEAttrNameExecutionPlanComputedIntermediateUids

+ const BEAttrNameExecutionPlanEngineConfig

+ const BEAttrNameExecutionPlanHandle

+ const BEAttrNameExecutionPlanRunOnlyIntermediateUids

+ const BEAttrNameExecutionPlanWorkspaceSize

+ const BEAttrNameIntermediateInfoDependentAttributes

+ const BEAttrNameIntermediateInfoDependentDataUids

+ const BEAttrNameIntermediateInfoSize

+ const BEAttrNameIntermediateInfoUniqueId

+ const BEAttrNameKnobChoiceKnobType

+ const BEAttrNameKnobChoiceKnobValue

+ const BEAttrNameKnobInfoMaximumValue

+ const BEAttrNameKnobInfoMinimumValue

+ const BEAttrNameKnobInfoStride

+ const BEAttrNameKnobInfoType

+ const BEAttrNameLayoutInfoTensorUid

+ const BEAttrNameLayoutInfoTypes

+ const BEAttrNameOperationConvolutionBwdDataAlpha

+ const BEAttrNameOperationConvolutionBwdDataBeta

+ const BEAttrNameOperationConvolutionBwdDataConvDesc

+ const BEAttrNameOperationConvolutionBwdDataDx

+ const BEAttrNameOperationConvolutionBwdDataDy

+ const BEAttrNameOperationConvolutionBwdDataW

+ const BEAttrNameOperationConvolutionBwdFilterAlpha

+ const BEAttrNameOperationConvolutionBwdFilterBeta

+ const BEAttrNameOperationConvolutionBwdFilterConvDesc

+ const BEAttrNameOperationConvolutionBwdFilterDw

+ const BEAttrNameOperationConvolutionBwdFilterDy

+ const BEAttrNameOperationConvolutionBwdFilterX

+ const BEAttrNameOperationConvolutionForwardAlpha

+ const BEAttrNameOperationConvolutionForwardBeta

+ const BEAttrNameOperationConvolutionForwardConvDesc

+ const BEAttrNameOperationConvolutionForwardW

+ const BEAttrNameOperationConvolutionForwardX

+ const BEAttrNameOperationConvolutionForwardY

+ const BEAttrNameOperationGenstatsMathPrec

+ const BEAttrNameOperationGenstatsMode

+ const BEAttrNameOperationGenstatsSqsumdesc

+ const BEAttrNameOperationGenstatsSumdesc

+ const BEAttrNameOperationGenstatsXdesc

+ const BEAttrNameOperationPointwiseAlpha1

+ const BEAttrNameOperationPointwiseAlpha2

+ const BEAttrNameOperationPointwiseBdesc

+ const BEAttrNameOperationPointwisePwDescriptor

+ const BEAttrNameOperationPointwiseXdesc

+ const BEAttrNameOperationPointwiseYdesc

+ const BEAttrNameOperationgraphEngineGlobalCount

+ const BEAttrNameOperationgraphHandle

+ const BEAttrNameOperationgraphOps

+ const BEAttrNamePointwiseMathPrec

+ const BEAttrNamePointwiseMode

+ const BEAttrNamePointwiseNanPropagation

+ const BEAttrNamePointwiseReluLowerClip

+ const BEAttrNamePointwiseReluUpperClip

+ const BEAttrNameTensorByteAlignment

+ const BEAttrNameTensorDataType

+ const BEAttrNameTensorDimensions

+ const BEAttrNameTensorIsVirtual

+ const BEAttrNameTensorStrides

+ const BEAttrNameTensorUniqueId

+ const BEAttrNameTensorVectorCount

+ const BEAttrNameTensorVectorizedDimension

+ const BEAttrNameVariantPackDataPointers

+ const BEAttrNameVariantPackIntermediates

+ const BEAttrNameVariantPackUniqueIds

+ const BEAttrNameVariantPackWorkspace

+ func (e BackendAttributeName) C() C.cudnnBackendAttributeName_t

+ func (e BackendAttributeName) String() string

+ type BackendAttributeType int

+ const BEAttrAttribName

+ const BEAttrBackendDescriptor

+ const BEAttrBoolean

+ const BEAttrConvolutionMode

+ const BEAttrDataType

+ const BEAttrDouble

+ const BEAttrFloat

+ const BEAttrGenstatsMode

+ const BEAttrHandle

+ const BEAttrHeurMode

+ const BEAttrInt64

+ const BEAttrKnobType

+ const BEAttrLayoutType

+ const BEAttrNanPropogation

+ const BEAttrNumericalNote

+ const BEAttrPointwiseMode

+ const BEAttrVoidPtr

+ func (e BackendAttributeType) C() C.cudnnBackendAttributeType_t

+ func (e BackendAttributeType) String() string

+ type BackendDescriptorType int

+ const BEDescriptorConvolutionDescriptor

+ const BEDescriptorEngineDescriptor

+ const BEDescriptorEnginecfgDescriptor

+ const BEDescriptorEngineheurDescriptor

+ const BEDescriptorExecutionPlanDescriptor

+ const BEDescriptorIntermediateInfoDescriptor

+ const BEDescriptorKnobChoiceDescriptor

+ const BEDescriptorKnobInfoDescriptor

+ const BEDescriptorLayoutInfoDescriptor

+ const BEDescriptorOperationConvolutionBackwardDataDescriptor

+ const BEDescriptorOperationConvolutionBackwardFilterDescriptor

+ const BEDescriptorOperationConvolutionForwardDescriptor

+ const BEDescriptorOperationGenStatsDescriptor

+ const BEDescriptorOperationPointwiseDescriptor

+ const BEDescriptorOperationgraphDescriptor

+ const BEDescriptorPointwiseDescriptor

+ const BEDescriptorTensorDescriptor

+ const BEDescriptorVariantPackDescriptor

+ func (e BackendDescriptorType) C() C.cudnnBackendDescriptorType_t

+ func (e BackendDescriptorType) String() string

+ type BackendHeurMode int

+ const Instant

+ const SCount

+ func (e BackendHeurMode) C() C.cudnnBackendHeurMode_t

+ func (e BackendHeurMode) String() string

+ type BackendKnobType int

+ const ChunkK

+ const Counts

+ const CtaSplitKMode

+ const Edge

+ const IdxMode

+ const Kblock

+ const Ldga

+ const Ldgb

+ const Ldgc

+ const Multiply

+ const ReductionMode

+ const Singlebuffer

+ const Sliced

+ const Specfilt

+ const SplitH

+ const SplitK

+ const SplitKBuf

+ const SplitKSlc

+ const SplitRs

+ const Stages

+ const Swizzle

+ const TileSize

+ const Tilek

+ const UseTex

+ const WinoTile

+ func (e BackendKnobType) C() C.cudnnBackendKnobType_t

+ func (e BackendKnobType) String() string

+ type BackendLayoutType int

+ const BELayoutCount

+ const BELayoutPreferredNchw

+ const BELayoutPreferredNhwc

+ const BELayoutPreferredPad4ck

+ const BELayoutPreferredPad8ck

+ func (e BackendLayoutType) C() C.cudnnBackendLayoutType_t

+ func (e BackendLayoutType) String() string

+ type BackendNumericalNote int

+ const DownConvertInputs

+ const Fft

+ const Nondeterministic

+ const ReducedPrecisionReduction

+ const TensorCore

+ const TypeCount

+ const Winograd

+ func (e BackendNumericalNote) C() C.cudnnBackendNumericalNote_t

+ func (e BackendNumericalNote) String() string

+ type BatchNormOps int

+ const BatchNorm

+ const BatchNormActivation

+ const BatchNormAddActivation

+ func (e BatchNormOps) C() C.cudnnBatchNormOps_t

+ func (e BatchNormOps) String() string

type Context

+ func (co *Context) FindConvolutionBackwardDataAlgorithm(wDesc *Filter, dyDesc *TensorDescriptor, convDesc *Convolution, ...) (returnedAlgoCount int, perfResults *ConvolutionBwdDataPerf, err error)

+ func (co *Context) FindConvolutionBackwardDataAlgorithmEx(wDesc *Filter, w Memory, dyDesc *TensorDescriptor, dy Memory, ...) (returnedAlgoCount int, perfResults *ConvolutionBwdDataPerf, err error)

+ func (co *Context) FindConvolutionBackwardFilterAlgorithm(xDesc *TensorDescriptor, dyDesc *TensorDescriptor, convDesc *Convolution, ...) (returnedAlgoCount int, perfResults *ConvolutionBwdPerf, err error)

+ func (co *Context) FindConvolutionBackwardFilterAlgorithmEx(xDesc *TensorDescriptor, x Memory, dyDesc *TensorDescriptor, y Memory, ...) (returnedAlgoCount int, perfResults *ConvolutionBwdPerf, err error)

+ func (co *Context) FindConvolutionForwardAlgorithm(xDesc *TensorDescriptor, wDesc *Filter, convDesc *Convolution, ...) (returnedAlgoCount int, perfResults *ConvolutionFwdPerf, err error)

+ func (co *Context) FindConvolutionForwardAlgorithmEx(xDesc *TensorDescriptor, x Memory, wDesc *Filter, w Memory, ...) (returnedAlgoCount int, perfResults *ConvolutionFwdPerf, err error)

type DataType

+ const Int8x32

+ const Uint8

+ const Uint8x4

+ type FoldingDirection int

+ const Fold

+ const Unfold

+ func (e FoldingDirection) C() C.cudnnFoldingDirection_t

+ func (e FoldingDirection) String() string

+ type ForwardMode int

+ const Inference

+ const Training

+ func (e ForwardMode) C() C.cudnnForwardMode_t

+ func (e ForwardMode) String() string

+ type FusedOpConsts struct

+ func NewFusedOpConsts(ops FusedOps, paramLabel FusedOpsConstParamLabel, param Memory) (retVal *FusedOpConsts, err error)

+ func (f *FusedOpConsts) C() C.cudnnFusedOpsConstParamPack_t

+ func (f *FusedOpConsts) Ops() FusedOps

+ func (f *FusedOpConsts) Param() Memory

+ func (f *FusedOpConsts) ParamLabel() FusedOpsConstParamLabel

+ type FusedOpVariantParams struct

+ func NewFusedOpVariantParams(ops FusedOps, paramLabel FusedOpsVariantParamLabel, ptr Memory) (retVal *FusedOpVariantParams, err error)

+ func (f *FusedOpVariantParams) C() C.cudnnFusedOpsVariantParamPack_t

+ func (f *FusedOpVariantParams) Ops() FusedOps

+ func (f *FusedOpVariantParams) ParamLabel() FusedOpsVariantParamLabel

+ func (f *FusedOpVariantParams) Ptr() Memory

+ type FusedOps int

+ const BnFinalizeStatisticsInference

+ const BnFinalizeStatisticsTraining

+ const ConvScaleBiasAddActivation

+ const DactivationForkDbatchnorm

+ const ScaleBiasActivationConvBnstats

+ const ScaleBiasActivationWgrad

+ const ScaleBiasAddActivationGenBitmask

+ func (e FusedOps) C() C.cudnnFusedOps_t

+ func (e FusedOps) String() string

+ type FusedOpsConstParamLabel int

+ const ActivationBitmaskDesc

+ const ActivationBitmaskPlaceholder

+ const ActivationDesc

+ const BnBiasPlaceholder

+ const BnDbiasPlaceholder

+ const BnDscalePlaceholder

+ const BnEqbiasPlaceholder

+ const BnEqscalePlaceholder

+ const BnEqscalebiasDesc

+ const BnMode

+ const BnRunningMeanPlaceholder

+ const BnRunningVarPlaceholder

+ const BnSavedInvstdPlaceholder

+ const BnSavedMeanPlaceholder

+ const BnScalePlaceholder

+ const BnScalebiasMeanvarDesc

+ const BnZEqbiasPlaceholder

+ const BnZEqscalePlaceholder

+ const BnZEqscalebiasDesc

+ const ConvDesc

+ const DwdataPlaceholder

+ const Dwdesc

+ const DxdataPlaceholder

+ const Dxdesc

+ const DydataPlaceholder

+ const Dydesc

+ const DzdataPlaceholder

+ const Dzdesc

+ const WdataPlaceholder

+ const Wdesc

+ const XdataPlaceholder

+ const Xdesc

+ const YdataPlaceholder

+ const Ydesc

+ const YsqsumPlaceholder

+ const YstatsDesc

+ const YsumPlaceholder

+ const ZdataPlaceholder

+ const Zdesc

+ func (e FusedOpsConstParamLabel) C() C.cudnnFusedOpsConstParamLabel_t

+ func (e FusedOpsConstParamLabel) String() string

+ type FusedOpsPointerPlaceHolder int

+ const NullPtr

+ const Ptr16

+ const PtrElemAligned

+ func (e FusedOpsPointerPlaceHolder) C() C.cudnnFusedOpsPointerPlaceHolder_t

+ func (e FusedOpsPointerPlaceHolder) String() string

+ type FusedOpsVariantParamLabel int

+ const PtrActivationBitmask

+ const PtrBnBias

+ const PtrBnDbias

+ const PtrBnDscale

+ const PtrBnEqbias

+ const PtrBnEqscale

+ const PtrBnRunningMean

+ const PtrBnRunningVar

+ const PtrBnSavedInvstd

+ const PtrBnSavedMean

+ const PtrBnScale

+ const PtrBnZEqbias

+ const PtrBnZEqscale

+ const PtrDwdata

+ const PtrDxdata

+ const PtrDydata

+ const PtrDzdata

+ const PtrWdata

+ const PtrWorkspace

+ const PtrXdata

+ const PtrYdata

+ const PtrYsqsum

+ const PtrYsum

+ const PtrZdata

+ const ScalarDoubleBnEpsilon

+ const ScalarDoubleBnExpAvgFactor

+ const ScalarInt64TBnAccumulationCount

+ const ScalarSizeTWorkspaceSizeInBytes

+ func (e FusedOpsVariantParamLabel) C() C.cudnnFusedOpsVariantParamLabel_t

+ func (e FusedOpsVariantParamLabel) String() string

+ type GenStatsMode int

+ const SumSq

+ func (e GenStatsMode) C() C.cudnnGenStatsMode_t

+ func (e GenStatsMode) String() string

type LRN

+ func (l *LRN) C() C.cudnnLRNDescriptor_t

+ type LossNormalizationMode int

+ const LossNormNone

+ const LossNormSoftmax

+ func (e LossNormalizationMode) C() C.cudnnLossNormalizationMode_t

+ func (e LossNormalizationMode) String() string

type MathType

+ const FmaMath

+ const TensorOpMathAllowConversion

+ type MultiHeadAttnWeightKind int

+ const KBiases

+ const KWeights

+ const OBiases

+ const OWeights

+ const QBiases

+ const QWeights

+ const VBiases

+ const VWeights

+ func (e MultiHeadAttnWeightKind) C() C.cudnnMultiHeadAttnWeightKind_t

+ func (e MultiHeadAttnWeightKind) String() string

+ type NormAlgo int

+ const PersistNorm

+ const StandardNorm

+ func (e NormAlgo) C() C.cudnnNormAlgo_t

+ func (e NormAlgo) String() string

+ type NormMode int

+ const NormPerActivation

+ const NormPerChannel

+ func (e NormMode) C() C.cudnnNormMode_t

+ func (e NormMode) String() string

+ type NormOps int

+ const Norm

+ const NormActivation

+ const NormAddActivation

+ func (e NormOps) C() C.cudnnNormOps_t

+ func (e NormOps) String() string

type OpTensorOp

+ const TensorAdd

+ const TensorMax

+ const TensorMin

+ const TensorMul

+ const TensorNot

+ const TensorSqrt

+ type PointwiseMode int

+ const PointwiseAdd

+ const PointwiseEluFwd

+ const PointwiseMax

+ const PointwiseMin

+ const PointwiseMul

+ const PointwiseReluFwd

+ const PointwiseSigmoidFwd

+ const PointwiseSqrt

+ const PointwiseTanhFwd

+ func (e PointwiseMode) C() C.cudnnPointwiseMode_t

+ func (e PointwiseMode) String() string

type RNNAlgo

+ const CountRNN

+ const PersistDynamicRNN

+ const PersistStaticRNN

+ const StandardRNN

+ type RNNBiasMode int

+ const RNNDoubleBias

+ const RNNNoBias

+ const RNNSingleInpBias

+ const RNNSingleRecBias

+ func (e RNNBiasMode) C() C.cudnnRNNBiasMode_t

+ func (e RNNBiasMode) String() string

+ type RNNClipMode int

+ const RNNClipMinmax

+ const RNNClipNone

+ func (e RNNClipMode) C() C.cudnnRNNClipMode_t

+ func (e RNNClipMode) String() string

+ type RNNData struct

+ func NewRNNData(dataType DataType, layout RNNDataLayout, maxSeqLength int, batchSize int, ...) (retVal *RNNData, err error)

+ func (r *RNNData) BatchSize() int

+ func (r *RNNData) C() C.cudnnRNNDataDescriptor_t

+ func (r *RNNData) DataType() DataType

+ func (r *RNNData) Layout() RNNDataLayout

+ func (r *RNNData) MaxSeqLength() int

+ func (r *RNNData) PaddingFill() Memory

+ func (r *RNNData) SeqLengthArray() []int

+ func (r *RNNData) VectorSize() int

+ type RNNDataLayout int

+ const BatchMajorUnpacked

+ const SeqMajorPacked

+ const SeqMajorUnpacked

+ func (e RNNDataLayout) C() C.cudnnRNNDataLayout_t

+ func (e RNNDataLayout) String() string

type Reduction

+ func (r *Reduction) C() C.cudnnReduceTensorDescriptor_t

+ type ReorderType int

+ const DefaultReorder

+ const NoReorder

+ func (e ReorderType) C() C.cudnnReorderType_t

+ func (e ReorderType) String() string

+ type SeqData struct

+ func NewSeqData(dataType DataType, nbDims int, dimA []int, axes []SeqDataAxis, ...) (retVal *SeqData, err error)

+ func (s *SeqData) Axes() []SeqDataAxis

+ func (s *SeqData) C() C.cudnnSeqDataDescriptor_t

+ func (s *SeqData) DataType() DataType

+ func (s *SeqData) DimA() []int

+ func (s *SeqData) NbDims() int

+ func (s *SeqData) PaddingFill() Memory

+ func (s *SeqData) SeqLengthArray() []int

+ func (s *SeqData) SeqLengthArraySize() uintptr

+ type SeqDataAxis int

+ const BatchDim

+ const BeamDim

+ const TimeDim

+ const VectDim

+ func (e SeqDataAxis) C() C.cudnnSeqDataAxis_t

+ func (e SeqDataAxis) String() string

+ type Severity int

+ const Error

+ const Fatal

+ const Info

+ const Warning

+ func (e Severity) C() C.cudnnSeverity_t

+ func (e Severity) String() string

type SpatialTransformer

+ func (s *SpatialTransformer) C() C.cudnnSpatialTransformerDescriptor_t

+ func (s *SpatialTransformer) DimA() []int

+ type Status = cudnnStatus

type TensorDescriptor

+ func (te *TensorDescriptor) DeriveBNTensorDescriptor(xDesc *TensorDescriptor, mode BatchNormMode) (derivedBnDesc *TensorDescriptor, err error)

+ type TensorTransform struct

+ func NewTensorTransform(nbDims uint32, destFormat TensorFormat, padBeforeA []int32, padAfterA []int32, ...) (retVal *TensorTransform, err error)

+ func (t *TensorTransform) C() C.cudnnTensorTransformDescriptor_t

+ func (t *TensorTransform) DestFormat() TensorFormat

+ func (t *TensorTransform) Direction() FoldingDirection

+ func (t *TensorTransform) FoldA() []uint32

+ func (t *TensorTransform) NDims() uint32

+ func (t *TensorTransform) PadAfterA() []int32

+ func (t *TensorTransform) PadBeforeA() []int32

+ type WgradMode int

+ const Set

+ func (e WgradMode) C() C.cudnnWgradMode_t

+ func (e WgradMode) String() string

v0.9.3

May 5, 2020

v0.9.2

Feb 12, 2020

v0.9.1

Sep 4, 2019

Changes in this version

+ const AllocFailed

+ const ArchMismatch

+ const BadParam

+ const BwdData

+ const BwdFilter

+ const ExecutionFailed

+ const Fwd

+ const InternalError

+ const InvalidValue

+ const LicenseError

+ const MappingError

+ const NotInitialized

+ const NotSupported

+ const RuntimeFpOverflow

+ const RuntimeInProgress

+ const RuntimePrerequisiteMissing

+ const Success

+ type Activation struct

+ func NewActivation(mode ActivationMode, reluNanOpt NanPropagation, coef float64) (retVal *Activation, err error)

+ func (a *Activation) Coef() float64

+ func (a *Activation) Mode() ActivationMode

+ func (a *Activation) ReluNanOpt() NanPropagation

+ type ActivationMode int

+ const ClippedReLU

+ const Elu

+ const ReLU

+ const Sigmoid

+ const Tanh

+ func (e ActivationMode) C() C.cudnnActivationMode_t

+ func (e ActivationMode) String() string

+ type BatchNormMode int

+ const PerActivation

+ const Spatial

+ const SpatialPersistent

+ func (e BatchNormMode) C() C.cudnnBatchNormMode_t

+ func (e BatchNormMode) String() string

+ type CTCLoss struct

+ func NewCTCLoss(compType DataType) (retVal *CTCLoss, err error)

+ func (c *CTCLoss) CompType() DataType

+ type CTCLossAlgo int

+ const DeterministicCTCLoss

+ const NonDeterministicCTCLoss

+ func (e CTCLossAlgo) C() C.cudnnCTCLossAlgo_t

+ func (e CTCLossAlgo) String() string

+ type Context struct

+ func NewContext() (retVal *Context)

+ func (co *Context) ActivationBackward(activationDesc *Activation, alpha float64, yDesc *TensorDescriptor, y Memory, ...) error

+ func (co *Context) ActivationForward(activationDesc *Activation, alpha float64, xDesc *TensorDescriptor, x Memory, ...) error

+ func (co *Context) AddTensor(alpha float64, aDesc *TensorDescriptor, A Memory, beta float64, ...) error

+ func (co *Context) BatchNormalizationBackward(mode BatchNormMode, alphaDataDiff float64, betaDataDiff float64, ...) error

+ func (co *Context) BatchNormalizationForwardInference(mode BatchNormMode, alpha float64, beta float64, xDesc *TensorDescriptor, ...) error

+ func (co *Context) BatchNormalizationForwardTraining(mode BatchNormMode, alpha float64, beta float64, xDesc *TensorDescriptor, ...) error

+ func (co *Context) CTCLoss(probsDesc *TensorDescriptor, probs Memory, labels []int, labelLengths []int, ...) error

+ func (co *Context) ConvolutionBackwardBias(alpha float64, dyDesc *TensorDescriptor, dy Memory, beta float64, ...) error

+ func (co *Context) ConvolutionBackwardData(alpha float64, wDesc *Filter, w Memory, dyDesc *TensorDescriptor, dy Memory, ...) error

+ func (co *Context) ConvolutionBackwardFilter(alpha float64, xDesc *TensorDescriptor, x Memory, dyDesc *TensorDescriptor, ...) error

+ func (co *Context) ConvolutionBiasActivationForward(alpha1 float64, xDesc *TensorDescriptor, x Memory, wDesc *Filter, w Memory, ...) error

+ func (co *Context) ConvolutionForward(alpha float64, xDesc *TensorDescriptor, x Memory, wDesc *Filter, w Memory, ...) error

+ func (co *Context) DivisiveNormalizationBackward(normDesc *LRN, mode DivNormMode, alpha float64, xDesc *TensorDescriptor, ...) error

+ func (co *Context) DivisiveNormalizationForward(normDesc *LRN, mode DivNormMode, alpha float64, xDesc *TensorDescriptor, ...) error

+ func (co *Context) DropoutBackward(dropoutDesc *Dropout, dydesc *TensorDescriptor, dy Memory, ...) error

+ func (co *Context) DropoutForward(dropoutDesc *Dropout, xdesc *TensorDescriptor, x Memory, ...) error

+ func (co *Context) DropoutGetStatesSize() (sizeInBytes uintptr, err error)

+ func (co *Context) GetRNNParamsSize(rnnDesc *RNN, xDesc *TensorDescriptor, dataType DataType) (sizeInBytes uintptr, err error)

+ func (co *Context) GetRNNTrainingReserveSize(rnnDesc *RNN, seqLength int, xDesc []*TensorDescriptor) (sizeInBytes uintptr, err error)

+ func (co *Context) GetRNNWorkspaceSize(rnnDesc *RNN, seqLength int, xDesc []*TensorDescriptor) (sizeInBytes uintptr, err error)

+ func (co *Context) GetReductionIndicesSize(reduceTensorDesc *Reduction, aDesc *TensorDescriptor, cDesc *TensorDescriptor) (sizeInBytes uintptr, err error)

+ func (co *Context) GetReductionWorkspaceSize(reduceTensorDesc *Reduction, aDesc *TensorDescriptor, cDesc *TensorDescriptor) (sizeInBytes uintptr, err error)

+ func (co *Context) Im2Col(xDesc *TensorDescriptor, x Memory, wDesc *Filter, convDesc *Convolution, ...) error

+ func (co *Context) LRNCrossChannelBackward(normDesc *LRN, lrnMode LRNMode, alpha float64, yDesc *TensorDescriptor, ...) error

+ func (co *Context) LRNCrossChannelForward(normDesc *LRN, lrnMode LRNMode, alpha float64, xDesc *TensorDescriptor, ...) error

+ func (co *Context) OpTensor(opTensorDesc *Op, alpha1 float64, aDesc *TensorDescriptor, A Memory, ...) error

+ func (co *Context) PoolingBackward(poolingDesc *Pooling, alpha float64, yDesc *TensorDescriptor, y Memory, ...) error

+ func (co *Context) PoolingForward(poolingDesc *Pooling, alpha float64, xDesc *TensorDescriptor, x Memory, ...) error

+ func (co *Context) RNNBackwardData(rnnDesc *RNN, seqLength int, yDesc []*TensorDescriptor, y Memory, ...) error

+ func (co *Context) RNNBackwardWeights(rnnDesc *RNN, seqLength int, xDesc []*TensorDescriptor, x Memory, ...) error

+ func (co *Context) RNNForwardInference(rnnDesc *RNN, seqLength int, xDesc []*TensorDescriptor, x Memory, ...) error

+ func (co *Context) RNNForwardTraining(rnnDesc *RNN, seqLength int, xDesc []*TensorDescriptor, x Memory, ...) error

+ func (co *Context) ReduceTensor(reduceTensorDesc *Reduction, indices Memory, indicesSizeInBytes uintptr, ...) error

+ func (co *Context) ScaleTensor(yDesc *TensorDescriptor, y Memory, alpha float64) error

+ func (co *Context) SoftmaxBackward(algo SoftmaxAlgorithm, mode SoftmaxMode, alpha float64, ...) error

+ func (co *Context) SoftmaxForward(algo SoftmaxAlgorithm, mode SoftmaxMode, alpha float64, ...) error

+ func (co *Context) SpatialTfGridGeneratorBackward(stDesc *SpatialTransformer, dgrid Memory, dtheta Memory) error

+ func (co *Context) SpatialTfGridGeneratorForward(stDesc *SpatialTransformer, theta Memory, grid Memory) error

+ func (co *Context) SpatialTfSamplerBackward(stDesc *SpatialTransformer, alpha float64, xDesc *TensorDescriptor, x Memory, ...) error

+ func (co *Context) SpatialTfSamplerForward(stDesc *SpatialTransformer, alpha float64, xDesc *TensorDescriptor, x Memory, ...) error

+ func (co *Context) TransformTensor(alpha float64, xDesc *TensorDescriptor, x Memory, beta float64, ...) error

+ func (ctx *Context) Close() error

+ func (ctx *Context) DoOp(op *Op, alpha1 float64, aDesc *TensorDescriptor, aData Memory, alpha2 float64, ...) error

+ func (handle *Context) NewRNN(hiddenSize int, numLayers int, dropout *Dropout, inputMode RNNInputMode, ...) (retVal *RNN, err error)

+ type Convolution struct

+ func NewConvolution(mathType MathType, groupCount int, padding, filterStride, dilation []int, ...) (retVal *Convolution, err error)

+ func (c *Convolution) CalcForwardOutputShape(input *TensorDescriptor, filter *Filter, dims int) (retVal []int, err error)

+ func (c *Convolution) Dilation() []int

+ func (c *Convolution) FilterStride() []int

+ func (c *Convolution) ForwardOutputShape(input *TensorDescriptor, filter *Filter, dims int) (retVal []int, err error)

+ func (c *Convolution) GroupCount() int

+ func (c *Convolution) MathType() MathType

+ func (c *Convolution) Padding() []int

+ type ConvolutionBwdDataAlgo int

+ const ConvolutionBwdDataAlgo0

+ const ConvolutionBwdDataAlgo1

+ const ConvolutionBwdDataAlgoCount

+ const ConvolutionBwdDataAlgoFFT

+ const ConvolutionBwdDataAlgoFFTTiling

+ const ConvolutionBwdDataAlgoWinograd

+ const ConvolutionBwdDataAlgoWinogradNonfused

+ func (c ConvolutionBwdDataAlgo) C() C.cudnnConvolutionBwdDataAlgo_t

+ type ConvolutionBwdDataPerf struct

+ Algo ConvolutionBwdDataAlgo

+ Determinism Determinism

+ Err error

+ MathType MathType

+ Memory uintptr

+ Time float64

+ func ConvolutionBwdDataPerfFromC(p C.cudnnConvolutionBwdDataAlgoPerf_t) *ConvolutionBwdDataPerf

+ type ConvolutionBwdFilterAlgo int

+ const ConvolutionBwdFilterAlgo0

+ const ConvolutionBwdFilterAlgo1

+ const ConvolutionBwdFilterAlgo3

+ const ConvolutionBwdFilterAlgoCount

+ const ConvolutionBwdFilterAlgoFFT

+ const ConvolutionBwdFilterAlgoFFTTiling

+ const ConvolutionBwdFilterAlgoWinograd

+ const ConvolutionBwdFilterAlgoWinogradNonfused

+ func (c ConvolutionBwdFilterAlgo) C() C.cudnnConvolutionBwdFilterAlgo_t

+ type ConvolutionBwdPerf struct

+ Algo ConvolutionBwdFilterAlgo

+ Determinism Determinism

+ Err error

+ MathType MathType

+ Memory uintptr

+ Time float64

+ type ConvolutionFwdAlgo int

+ const ConvolutionFwdAlgoCount

+ const ConvolutionFwdAlgoDirect

+ const ConvolutionFwdAlgoFFT

+ const ConvolutionFwdAlgoFFTTiling

+ const ConvolutionFwdAlgoGemm

+ const ConvolutionFwdAlgoImplicitGemm

+ const ConvolutionFwdAlgoImplicitPrecompGemm

+ const ConvolutionFwdAlgoWinograd

+ const ConvolutionFwdAlgoWinogradNonfused

+ func (c ConvolutionFwdAlgo) C() C.cudnnConvolutionFwdAlgo_t

+ type ConvolutionFwdPerf struct

+ Algo ConvolutionFwdAlgo

+ Determinism Determinism

+ Err error

+ MathType MathType

+ Memory uintptr

+ Time float64

+ type ConvolutionMode int

+ const CrossCorrelation

+ const StandardConvolution

+ func (e ConvolutionMode) C() C.cudnnConvolutionMode_t

+ type ConvolutionPreference byte

+ const NoWorkspace

+ const PreferFastest

+ const SpecifyWorkspaceLimit

+ func MakeConvolutionPreference(t ConvolutionType, pref ConvolutionPreference) ConvolutionPreference

+ func (c ConvolutionPreference) C() C.int

+ func (c ConvolutionPreference) IsBwd() bool

+ func (c ConvolutionPreference) IsData() bool

+ func (c ConvolutionPreference) IsFilter() bool

+ func (c ConvolutionPreference) IsFwd() bool

+ func (c ConvolutionPreference) Pref() ConvolutionPreference

+ type ConvolutionType byte

+ type DataType int

+ const Double

+ const Float

+ const Half

+ const Int32

+ const Int8

+ const Int8x4

+ func (e DataType) C() C.cudnnDataType_t

+ func (e DataType) String() string

+ type Determinism int

+ const Deterministic

+ const NonDeterministic

+ func (e Determinism) C() C.cudnnDeterminism_t

+ func (e Determinism) String() string

+ type DirectionMode int

+ const Bidirectional

+ const Unidirectional

+ func (e DirectionMode) C() C.cudnnDirectionMode_t

+ func (e DirectionMode) String() string

+ type DivNormMode int

+ const PrecomputedMeans

+ func (e DivNormMode) C() C.cudnnDivNormMode_t

+ func (e DivNormMode) String() string

+ type Dropout struct

+ func NewDropout(dropout float64) (retVal *Dropout, err error)

+ func NewDropoutWithContext(dropout float64, handle *Context, states Memory, stateSizeInBytes uintptr, ...) (retVal *Dropout, err error)

+ func (d *Dropout) Dropout() float32

+ func (d *Dropout) Handle() *Context

+ func (d *Dropout) IsReady() bool

+ func (d *Dropout) RequiredStateSize(ctx *Context) (uintptr, error)

+ func (d *Dropout) Reset()

+ func (d *Dropout) Seed() uint64

+ func (d *Dropout) StateSizeInBytes() uintptr

+ func (d *Dropout) States() Memory

+ func (d *Dropout) Use(ctx *Context, states Memory, stateSizeInBytes uintptr, seed uint64) error

+ func (dr *Dropout) RestoreDropoutDescriptor(handle *Context, dropout float32, states Memory, stateSizeInBytes uintptr, ...) error

+ type ErrQueryMode int

+ const Blocking

+ const Nonblocking

+ const Rawcode

+ func (e ErrQueryMode) C() C.cudnnErrQueryMode_t

+ func (e ErrQueryMode) String() string

+ type Filter struct

+ func NewFilter(dataType DataType, format TensorFormat, shape []int) (retVal *Filter, err error)

+ func (f *Filter) DataType() DataType

+ func (f *Filter) Format() TensorFormat

+ func (f *Filter) Shape() []int

+ type IndicesType int

+ const Indices16

+ const Indices32

+ const Indices64

+ const Indices8

+ func (e IndicesType) C() C.cudnnIndicesType_t

+ func (e IndicesType) String() string

+ type LRN struct

+ func NewLRN(lrnN uint, lrnAlpha float64, lrnBeta float64, lrnK float64) (retVal *LRN, err error)

+ func (l *LRN) LrnAlpha() float64

+ func (l *LRN) LrnBeta() float64

+ func (l *LRN) LrnK() float64

+ func (l *LRN) LrnN() uint

+ type LRNMode int

+ const CrossChannelDim1

+ func (e LRNMode) C() C.cudnnLRNMode_t

+ func (e LRNMode) String() string

+ type MathType int

+ const DefaultMath

+ const TensorOpMath

+ func (e MathType) C() C.cudnnMathType_t

+ func (e MathType) String() string

+ type Memory interface

+ IsNativelyAccessible func() bool

+ Pointer func() unsafe.Pointer

+ Uintptr func() uintptr

+ type NanPropagation int

+ const NotPropagateNan

+ const PropagateNan

+ func (e NanPropagation) C() C.cudnnNanPropagation_t

+ func (e NanPropagation) String() string

+ type Op struct

+ func NewOp(op OpTensorOp, dt DataType, prop NanPropagation) (*Op, error)

+ func (op *Op) DataType() DataType

+ func (op *Op) NaNPropagation() NanPropagation

+ func (op *Op) Op() OpTensorOp

+ type OpTensorOp int

+ const Add

+ const Max

+ const Min

+ const Mul

+ const Not

+ const Sqrt

+ func (e OpTensorOp) C() C.cudnnOpTensorOp_t

+ func (e OpTensorOp) String() string

+ type Pooling struct

+ func NewPooling(mode PoolingMode, maxpoolingNanOpt NanPropagation, ...) (retVal *Pooling, err error)

+ func (p *Pooling) CalcOutputShape(input *TensorDescriptor, dims int) (retVal []int, err error)

+ func (p *Pooling) Mode() PoolingMode

+ func (p *Pooling) NaNPropagation() NanPropagation

+ func (p *Pooling) OutputShape(input *TensorDescriptor, dims int) (retVal []int, err error)

+ func (p *Pooling) Padding() []int

+ func (p *Pooling) Shape() []int

+ func (p *Pooling) Strides() []int

+ type PoolingMode int

+ const AverageCountExcludePadding

+ const AverageCountIncludePadding

+ const MaxDeterministic

+ const MaxPooling

+ func (e PoolingMode) C() C.cudnnPoolingMode_t

+ func (e PoolingMode) String() string

+ type RNN struct

+ func (r *RNN) Algo() RNNAlgo

+ func (r *RNN) DataType() DataType

+ func (r *RNN) DirectionMode() DirectionMode

+ func (r *RNN) Dropout() *Dropout

+ func (r *RNN) HiddenSize() int

+ func (r *RNN) InputMode() RNNInputMode

+ func (r *RNN) MatrixMathType() MathType

+ func (r *RNN) Mode() RNNMode

+ func (r *RNN) NumLayers() int

+ func (r *RNN) SetMatrixMathType(mathType MathType) error

+ type RNNAlgo int

+ const PersistDynamic

+ const PersistStatic

+ const Standard

+ func (e RNNAlgo) C() C.cudnnRNNAlgo_t

+ func (e RNNAlgo) String() string

+ type RNNInputMode int

+ const LinearInput

+ const SkipInput

+ func (e RNNInputMode) C() C.cudnnRNNInputMode_t

+ func (e RNNInputMode) String() string

+ type RNNMode int

+ const GRU

+ const LSTM

+ const RNNReLU

+ const RNNTanh

+ func (e RNNMode) C() C.cudnnRNNMode_t

+ func (e RNNMode) String() string

+ type RNNPlan struct

+ func NewRNNPlan(minibatch int, dataType DataType, desc *RNN) (plan *RNNPlan, err error)

+ func (p *RNNPlan) Update(rnn *RNN) error

+ type ReduceTensorIndices int

+ const ReduceFlattenedIndices

+ const ReduceNoIndices

+ func (e ReduceTensorIndices) C() C.cudnnReduceTensorIndices_t

+ func (e ReduceTensorIndices) String() string

+ type ReduceTensorOp int

+ const ReduceAdd

+ const ReduceAmax

+ const ReduceAvg

+ const ReduceMax

+ const ReduceMin

+ const ReduceMul

+ const ReduceMulNoZeros

+ const ReduceNorm1

+ const ReduceNorm2

+ func (e ReduceTensorOp) C() C.cudnnReduceTensorOp_t

+ func (e ReduceTensorOp) String() string

+ type Reduction struct

+ func NewReduction(reduceTensorOp ReduceTensorOp, reduceTensorCompType DataType, ...) (retVal *Reduction, err error)

+ func (r *Reduction) ReduceTensorCompType() DataType

+ func (r *Reduction) ReduceTensorIndices() ReduceTensorIndices

+ func (r *Reduction) ReduceTensorIndicesType() IndicesType

+ func (r *Reduction) ReduceTensorNanOpt() NanPropagation

+ func (r *Reduction) ReduceTensorOp() ReduceTensorOp

+ type SamplerType int

+ const Bilinear

+ func (e SamplerType) C() C.cudnnSamplerType_t

+ func (e SamplerType) String() string

+ type SoftmaxAlgorithm int

+ const Accurate

+ const Fast

+ const Log

+ func (e SoftmaxAlgorithm) C() C.cudnnSoftmaxAlgorithm_t

+ func (e SoftmaxAlgorithm) String() string

+ type SoftmaxMode int

+ const Channel

+ const Instance

+ func (e SoftmaxMode) C() C.cudnnSoftmaxMode_t

+ func (e SoftmaxMode) String() string

+ type SpatialTransformer struct

+ func NewSpatialTransformer(samplerType SamplerType, dataType DataType, nbDims int, dimA []int) (retVal *SpatialTransformer, err error)

+ func (s *SpatialTransformer) DataType() DataType

+ func (s *SpatialTransformer) NbDims() int

+ func (s *SpatialTransformer) SamplerType() SamplerType

+ type TensorDescriptor struct

+ func NewTensorDescriptor(format TensorFormat, dt DataType, shape, strides []int) (*TensorDescriptor, error)

+ func (t *TensorDescriptor) DataType() DataType

+ func (t *TensorDescriptor) Format() TensorFormat

+ func (t *TensorDescriptor) Shape() []int

+ func (t *TensorDescriptor) Strides() []int

+ func (te *TensorDescriptor) DropoutGetReserveSpaceSize() (sizeInBytes uintptr, err error)

+ type TensorFormat int

+ const NCHW

+ const NCHWVectC

+ const NHWC

+ func (e TensorFormat) C() C.cudnnTensorFormat_t

+ func (e TensorFormat) String() string

v0.9.0-working

Aug 18, 2018

v0.9.0-beta

Aug 12, 2018

?	: This menu
/	: Search site
f or F	: Jump to
y or Y	: Canonical URL