网络计算机amd,amd vs. nvidia 深度神经网络库api对照,已无敌。

amd versionoriginial

AMD ROCm MIOPENNVIDIA CUDA CUDNN

Handle

miopenStatus_tcudnnStatus_t

miopenCreatecudnnCreate

miopenCreateWithStreamN/A

miopenDestroycudnnDestroy

miopenSetStreamcudnnSetStream

miopenGetStreamcudnnGetStream

miopenGetKernelTimeN/A

miopenEnableProfilingN/A

Tensors

miopenDataType_tcudnnDataType_t

miopenTensorOp_tcudnnOpTensorOp_t

miopenCreateTensorDescriptorcudnnCreateOpTensorDescriptor

miopenSet4dTensorDescriptorcudnnSetTensor4dDescriptor

miopenGet4dTensorDescriptorcudnnGetTensor4dDescriptor

miopenSetTensorDescriptorcudnnSetOpTensorDescriptor

miopenGetTensorDescriptorSizecudnnGetTensorSizeInBytes

miopenGetTensorDescriptorcudnnGetOpTensorDescriptor

miopenDestroyTensorDescriptorcudnnDestroyOpTensorDescriptor

miopenOpTensorcudnnOpTensor

miopenSetTensorcudnnSetTensor

miopenScaleTensorcudnnScaleTensor

Activation Layers

miopenActivationMode_tcudnnActivationMode_t

miopenCreateActivationDescriptorcudnnCreateActivationDescriptor

miopenSetActivationDescriptorcudnnSetActivationDescriptor

miopenGetActivationDescriptorcudnnGetActivationDescriptor

miopenActivationForwardcudnnActivationForward

miopenActivationBackwardcudnnActivationBackward

miopenDestroyActivationDescriptorcudnnDestroyActivationDescriptor

Convolutional Layer

miopenConvolutionMode_tcudnnConvolutionMode_t

miopenConvFwdAlgorithm_tcudnnConvolutionFwdAlgo_t

miopenConvBwdWeightsAlgorithm_tcudnnConvolutionBwdFilterAlgo_t

miopenConvBwdDataAlgorithm_tcudnnConvolutionBwdDataAlgo_t

miopenConvAlgoPerf_tcudnnConvolutionFwdAlgoPerf_t / cudnnConvolutionBwdFilterAlgoPerf_t

miopenCreateConvolutionDescriptorcudnnCreateConvolutionDescriptor

miopenInitConvolutionDescriptorN/A

miopenGetConvolutionDescriptorcudnnSetConvolution2dDescriptor / cudnnGetConvolutionNdDescriptor

miopenGetConvolutionForwardOutputDimcudnnGetConvolution2dForwardOutputDim /  cudnnGetConvolutionNdForwardOutputDim

miopenConvolutionForwardGetWorkSpaceSizecudnnGetConvolutionForwardWorkspaceSize

miopenFindConvolutionForwardAlgorithmcudnnFindConvolutionForwardAlgorithm /  cudnnFindConvolutionForwardAlgorithmEx

miopenConvolutionForwardcudnnConvolutionForward

miopenConvolutionForwardBiascudnnConvolutionBiasActivationForward

miopenFindConvolutionBackwardDataAlgorithmcudnnFindConvolutionBackwardDataAlgorithm /  cudnnFindConvolutionBackwardDataAlgorithmEx

miopenConvolutionBackwardDatacudnnConvolutionBackwardData

miopenConvolutionBackwardDataGetWorkSpaceSizecudnnGetConvolutionBackwardFilterWorkspaceSize

miopenConvolutionBackwardWeightsGetWorkSpaceSizecudnnGetConvolutionBackwardDataWorkspaceSize

miopenFindConvolutionBackwardWeightsAlgorithmcudnnFindConvolutionBackwardFilterAlgorithm /  cudnnFindConvolutionBackwardFilterAlgorithmEx

miopenConvolutionBackwardWeightscudnnConvolutionBackwardFilter

miopenConvolutionBackwardBiascudnnConvolutionBackwardBias

miopenDestroyConvolutionDescriptorcudnnDestroyConvolutionDescriptor

Pooling Layer

miopenPoolingMode_tcudnnPoolingMode_t

miopenCreatePoolingDescriptorcudnnCreatePoolingDescriptor

miopenSet2dPoolingDescriptorcudnnSetPooling2dDescriptor / cudnnSetPoolingNdDescriptor

miopenGet2dPoolingDescriptorcudnnGetPooling2dDescriptor / cudnnGetPoolingNdDescriptor

miopenGetPoolingForwardOutputDimcudnnGetPooling2dForwardOutputDim / cudnnGetPoolingNdForwardOutputDim

miopenPoolingGetWorkSpaceSizecudnnDestroyPoolingDescriptor

miopenPoolingForwardcudnnPoolingForward

miopenPoolingBackwardcudnnPoolingBackward

miopenDestroyPoolingDescriptorcudnnDestroyPoolingDescriptor

Softmax Layer

miopenSoftmaxForwardcudnnSoftmaxForward

miopenSoftmaxBackwardcudnnSoftmaxBackward

Local Response Normalization Layer

miopenLRNMode_tcudnnLRNMode_t

miopenCreateLRNDescriptorcudnnCreateLRNDescriptor

miopenSetLRNDescriptorcudnnSetLRNDescriptor

miopenGetLRNDescriptorcudnnGetLRNDescriptor

miopenLRNGetWorkSpaceSizeN/A

miopenLRNForwardcudnnLRNCrossChannelForward / cudnnDivisiveNormalizationForward

miopenLRNBackwardcudnnLRNCrossChannelBackward / cudnnDivisiveNormalizationBackward

miopenDestroyLRNDescriptorcudnnDestroyLRNDescriptor

Batch Normalization Layer

miopenBatchNormMode_tcudnnBatchNormMode_t

miopenDeriveBNTensorDescriptorcudnnDeriveBNTensorDescriptor

miopenBatchNormalizationForwardTrainingcudnnBatchNormalizationForwardTraining

miopenBatchNormalizationForwardInferencecudnnBatchNormalizationForwardInference

miopenBatchNormalizationBackwardcudnnBatchNormalizationBackward

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值