Esempio 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')
Esempio 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')
Esempio 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')
Esempio 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')