MCPcopy Create free account
hub / github.com/QWTforGithub/T2LDM / get_padding_value

Function get_padding_value

timm/models/layers/padding.py:36–56  ·  view source on GitHub ↗
(padding, kernel_size, **kwargs)

Source from the content-addressed store, hash-verified

34
35
36def get_padding_value(padding, kernel_size, **kwargs) -> Tuple[Tuple, bool]:
37 dynamic = False
38 if isinstance(padding, str):
39 # for any string padding, the padding will be calculated for you, one of three ways
40 padding = padding.lower()
41 if padding == 'same':
42 # TF compatible 'SAME' padding, has a performance and GPU memory allocation impact
43 if is_static_pad(kernel_size, **kwargs):
44 # static case, no extra overhead
45 padding = get_padding(kernel_size, **kwargs)
46 else:
47 # dynamic 'SAME' padding, has runtime/GPU memory overhead
48 padding = 0
49 dynamic = True
50 elif padding == 'valid':
51 # 'VALID' padding, same as padding=0
52 padding = 0
53 else:
54 # Default to PyTorch style 'same'-ish symmetric padding
55 padding = get_padding(kernel_size, **kwargs)
56 return padding, dynamic

Callers 5

__init__Method · 0.85
__init__Method · 0.85
create_pool2dFunction · 0.85
create_conv2d_padFunction · 0.85
__init__Method · 0.85

Calls 2

is_static_padFunction · 0.85
get_paddingFunction · 0.70

Tested by

no test coverage detected