bosk.block.zoo.multi_grained_scanning._pooling_impl#

Module Contents#

Functions#

_njit_max_pooling_1d(xs, result, kernel_size, stride, ...)

_njit_mean_pooling_1d(xs, result, kernel_size, stride, ...)

_njit_max_pooling_2d(xs, result, kernel_size, stride, ...)

_njit_mean_pooling_2d(xs, result, kernel_size, stride, ...)

bosk.block.zoo.multi_grained_scanning._pooling_impl._njit_max_pooling_1d(xs, result, kernel_size, stride, dilation)#
Parameters:
  • xs (numpy.ndarray) –

  • result (numpy.ndarray) –

  • kernel_size (Tuple[int]) –

  • stride (Tuple[int]) –

  • dilation (int) –

bosk.block.zoo.multi_grained_scanning._pooling_impl._njit_mean_pooling_1d(xs, result, kernel_size, stride, dilation)#
Parameters:
  • xs (numpy.ndarray) –

  • result (numpy.ndarray) –

  • kernel_size (Tuple[int]) –

  • stride (Tuple[int]) –

  • dilation (int) –

bosk.block.zoo.multi_grained_scanning._pooling_impl._njit_max_pooling_2d(xs, result, kernel_size, stride, dilation)#
Parameters:
  • xs (numpy.ndarray) –

  • result (numpy.ndarray) –

  • kernel_size (Tuple[int, int]) –

  • stride (Tuple[int, int]) –

  • dilation (int) –

bosk.block.zoo.multi_grained_scanning._pooling_impl._njit_mean_pooling_2d(xs, result, kernel_size, stride, dilation)#
Parameters:
  • xs (numpy.ndarray) –

  • result (numpy.ndarray) –

  • kernel_size (Tuple[int, int]) –

  • stride (Tuple[int, int]) –

  • dilation (int) –