Function _nearest_sample

python/tvm/topi/testing/grid_sample_python.py:109–119

()

        return (y, x)

    def _nearest_sample():
        for _b in range(batch):
            for _c in range(channel):
                for _h in range(out_height):
                    for _w in range(out_width):
                        y, x = _compute_source_index(_b, _h, _w)
                        # python round is not used here,
                        # beacause it is done toward the even choice
                        new_y = int(y + 0.5) if y > 0 else int(y - 0.5)
                        new_x = int(x + 0.5) if x > 0 else int(x - 0.5)
                        out[_b, _c, _h, _w] = _get_pixel(_b, _c, new_y, new_x)

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