Python get_pool2d_descの例

コード例 #1

ファイル: pooling_2d.py プロジェクト: 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')

コード例 #2

ファイル: pooling_2d.py プロジェクト: 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')

コード例 #3

ファイル: pooling_2d.py プロジェクト: 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')

コード例 #4

ファイル: pooling_2d.py プロジェクト: 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')