Exemplo n.º 1
0
 def create_pool_desc(self):
     return cudnn.get_pool2d_desc(
         (self.kh, self.kw), (self.sy, self.sx), (self.ph, self.pw),
         'CUDNN_POOLING_AVERAGE_COUNT_INCLUDE_PADDING')
Exemplo n.º 2
0
 def create_pool_desc(self):
     return cudnn.get_pool2d_desc(
         (self.kh, self.kw), (self.sy, self.sx), (self.ph, self.pw),
         'CUDNN_POOLING_AVERAGE_COUNT_INCLUDE_PADDING')
Exemplo n.º 3
0
 def create_pool_desc(self):
     return cudnn.get_pool2d_desc((self.kh, self.kw), (self.sy, self.sx),
                                  (self.ph, self.pw), 'CUDNN_POOLING_MAX')
Exemplo n.º 4
0
 def create_pool_desc(self):
     return cudnn.get_pool2d_desc(
         (self.kh, self.kw), (self.sy, self.sx), (self.ph, self.pw),
         'CUDNN_POOLING_MAX')