Function func

imperative/python/test/unit/xla/functional/test_xla_indexing.py:26–31

(inp, dout)

        @jit.xla_trace(without_host=True, capture_as_const=True)
        def func(inp, dout):
            gm.attach([inp])
            with gm:
                out = inp[index]
                gm.backward(out, dout)
            return out, inp.grad

        mge_rsts = func(inp, dout)
        xla_rsts = func(inp, dout)