(m)
| 2298 | |
| 2299 | |
| 2300 | def _to_torch_resample_padding_mode(m): |
| 2301 | ret = look_up_option(str(m), GridSamplePadMode, default=None) |
| 2302 | if ret is not None: |
| 2303 | return ret |
| 2304 | _mapping = { |
| 2305 | NdimageMode.CONSTANT: GridSamplePadMode.ZEROS, |
| 2306 | NdimageMode.GRID_CONSTANT: GridSamplePadMode.ZEROS, |
| 2307 | NdimageMode.NEAREST: GridSamplePadMode.BORDER, |
| 2308 | NdimageMode.REFLECT: GridSamplePadMode.REFLECTION, |
| 2309 | NdimageMode.WRAP: GridSamplePadMode.REFLECTION, |
| 2310 | NdimageMode.GRID_WRAP: GridSamplePadMode.REFLECTION, |
| 2311 | NdimageMode.GRID_MIRROR: GridSamplePadMode.REFLECTION, |
| 2312 | } |
| 2313 | ret = look_up_option(str(m), _mapping, default=None) |
| 2314 | if ret is not None: |
| 2315 | return ret |
| 2316 | return look_up_option(str(m), list(_mapping) + list(GridSamplePadMode)) |
| 2317 | |
| 2318 | |
| 2319 | @lru_cache(None) |
no test coverage detected
searching dependent graphs…