Searched refs:xla_result_handlers (Results 1 – 1 of 1) sorted by relevance
110 return xla_result_handlers[type(aval)](device, aval)121 xla_result_handlers: Dict[Type[core.AbstractValue], Callable[..., Callable]] = { variable1005 xla_result_handlers[AbstractToken] = lambda _, __: lambda _: token