Function _compute_source_index

python/tvm/topi/testing/grid_sample_python.py:93–107

(b, h, w)

        return x

    def _compute_source_index(b, h, w):
        y = grid[b, 1, h, w]
        x = grid[b, 0, h, w]
        y, x = _unnormalize(y, x)

        if padding_mode == "reflection":
            y = _reflect_coordinates(y, in_height)
            x = _reflect_coordinates(x, in_width)
            y = _clip_coordinates(y, in_height)
            x = _clip_coordinates(x, in_width)
        elif padding_mode == "border":
            y = _clip_coordinates(y, in_height)
            x = _clip_coordinates(x, in_width)

        return (y, x)

    def _nearest_sample():
        for _b in range(batch):