예제 #1
0
 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();
 }
예제 #2
0
 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();
 }