Esempi in Python per get_pool2d_desc

Esempio n. 1

File: pooling_2d.py Progetto: woodshop/complex-chainer

 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

File: pooling_2d.py Progetto: ALEXGUOQ/chainer

 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

File: pooling_2d.py Progetto: woodshop/complex-chainer

 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

File: pooling_2d.py Progetto: ALEXGUOQ/chainer

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