public void AddBias(CUDNNTensorDescriptor dstTensorDesc, Layer layer, int c, DeviceMemory<float> data) { _biasTensorDesc.Set4D(TensorFormat, DataType, 1, c, 1, 1); const float alpha = 1.0f; const float beta = 1.0f; _cudnn.AddTensor(CUDNNInterop.cudnnAddMode_t.CUDNN_ADD_SAME_C, alpha, _biasTensorDesc, layer.BiasD.Ptr, beta, dstTensorDesc, data.Ptr); }
public void AddBias(CUDNNTensorDescriptor dstTensorDesc, Layer layer, int c, DeviceMemory <float> data) { _biasTensorDesc.Set4D(TensorFormat, DataType, 1, c, 1, 1); const float alpha = 1.0f; const float beta = 1.0f; _cudnn.AddTensor(CUDNNInterop.cudnnAddMode_t.CUDNN_ADD_SAME_C, alpha, _biasTensorDesc, layer.BiasD.Ptr, beta, dstTensorDesc, data.Ptr); }
public Network(Worker worker) { _worker = worker; _cudnn = new CUDNN(_worker); _cublas = new CUBLAS(_worker); _srcTensorDesc = new CUDNNTensorDescriptor(); _dstTensorDesc = new CUDNNTensorDescriptor(); _biasTensorDesc = new CUDNNTensorDescriptor(); _filterDesc = new CUDNNFilterDescriptor(); _convDesc = new CUDNNConvolutionDescriptor(); _poolingDesc = new CUDNNPoolingDescriptor(); }