[XLA:CPU] Add support for CustomCall targets that return tuples.
Populate the tuple index table of the return value; the callee cannot do this since it does not know the buffer assignments. Explicitly enable custom_call_test only for cpu in the BUILD file, rather than disabling it on non-CPU backends. These tests would not work on any non-CPU backend. PiperOrigin-RevId: 225048065
Showing
想要评论请 注册 或 登录