Ejemplos de CUDNNTensorDescriptor en C# (CSharp)

Ejemplo n.º 1

Archivo: Network.cs Proyecto: magoroku15/AleaGPUTutorial

 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);
 }

Ejemplo n.º 2

        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);
        }

Ejemplo n.º 3

 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();
 }

Ejemplo n.º 4

Archivo: Network.cs Proyecto: magoroku15/AleaGPUTutorial

 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();
 }