fix(api): keep kernel slices within bounds
This commit is contained in:
parent
c1a4484b75
commit
5713957026
|
@ -79,8 +79,8 @@ def fix_node_name(key: str):
|
||||||
|
|
||||||
def kernel_slice(x: int, y: int, shape: Tuple[int, int, int, int]) -> Tuple[int, int]:
|
def kernel_slice(x: int, y: int, shape: Tuple[int, int, int, int]) -> Tuple[int, int]:
|
||||||
return (
|
return (
|
||||||
max(x, shape[2]),
|
min(x, shape[2] - 1),
|
||||||
max(y, shape[3]),
|
min(y, shape[3] - 1),
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue