Coverage for src/flag_gems/runtime/backend/_spacemit/ops/thnn_conv2d.py: 0%

1import logging 

2 

3from .conv2d import conv2d 

4 

5logger = logging.getLogger(__name__) 

6 

7 

8def thnn_conv2d_impl( 

9 input, weight, kernel_size=0, bias=None, stride=1, padding=0, groups=1 

10): 

11 logger.debug("GEMS_SPACEMIT THNN_CONV2D") 

12 dilation = 1 

13 return conv2d(input, weight, bias, padding, stride, dilation, groups) 

14 

15 

16def thnn_conv2d(input, weight, kernel_size=0, bias=None, stride=1, padding=0, groups=1): 

17 return thnn_conv2d_impl(input, weight, kernel_size, bias, stride, padding, groups)