Function __reflect

python/tvm/topi/testing/grid_sample_python.py:74–79

(index, size, corner_start)

    def _reflect_coordinates(i, size):
        def __refelection(i, size, corner_start):
            def __reflect(index, size, corner_start):
                index_align_corner = abs(corner_start - index)
                size_times = index_align_corner // size
                even = size_times % 2 == 0
                extra = index_align_corner - size_times * size
                return extra + corner_start if even else size - extra + corner_start

            if corner_start <= i <= size + corner_start:
                new_i = i