Helper function for finding shapes and chunks of end pads.
(array, pad_width, axes, mode)
| 971 | |
| 972 | |
| 973 | def get_pad_shapes_chunks(array, pad_width, axes, mode): |
| 974 | """ |
| 975 | Helper function for finding shapes and chunks of end pads. |
| 976 | """ |
| 977 | |
| 978 | pad_shapes = [list(array.shape), list(array.shape)] |
| 979 | pad_chunks = [list(array.chunks), list(array.chunks)] |
| 980 | |
| 981 | for d in axes: |
| 982 | for i in range(2): |
| 983 | pad_shapes[i][d] = pad_width[d][i] |
| 984 | if mode != "constant" or pad_width[d][i] == 0: |
| 985 | pad_chunks[i][d] = (pad_width[d][i],) |
| 986 | else: |
| 987 | pad_chunks[i][d] = normalize_chunks( |
| 988 | (max(pad_chunks[i][d]),), (pad_width[d][i],) |
| 989 | )[0] |
| 990 | |
| 991 | pad_shapes = [tuple(s) for s in pad_shapes] |
| 992 | pad_chunks = [tuple(c) for c in pad_chunks] |
| 993 | |
| 994 | return pad_shapes, pad_chunks |
| 995 | |
| 996 | |
| 997 | def linear_ramp_chunk(start, stop, num, dim, step): |
no test coverage detected
searching dependent graphs…