Exemplo n.º 1
0
 def add_tensor(handle, alpha, biasDesc, biasData, beta, srcDestDesc, srcDestData):
     cudnn.addTensor_v2(handle, cudnn.CUDNN_ADD_SAME_C, alpha, biasDesc, biasData, beta, srcDestDesc, srcDestData)
Exemplo n.º 2
0
 def add_tensor(handle, alpha, biasDesc, biasData, beta, srcDestDesc,
                srcDestData):
     cudnn.addTensor_v2(handle, cudnn.CUDNN_ADD_SAME_C, alpha, biasDesc,
                        biasData, beta, srcDestDesc, srcDestData)