(
spatial_dims: int, in_channels: int, out_channels: int, kernel_size: int = 3, stride: int = 1, bias: bool = False
)
| 20 | |
| 21 | |
| 22 | def get_conv_layer( |
| 23 | spatial_dims: int, in_channels: int, out_channels: int, kernel_size: int = 3, stride: int = 1, bias: bool = False |
| 24 | ): |
| 25 | return Convolution( |
| 26 | spatial_dims, in_channels, out_channels, strides=stride, kernel_size=kernel_size, bias=bias, conv_only=True |
| 27 | ) |
| 28 | |
| 29 | |
| 30 | def get_upsample_layer( |
no test coverage detected
searching dependent graphs…