Where does the retrieval done?

[sutakori]

Hi author, thanks for the great work. I have a problem about retrieval.

In paper, the retrieval is done in the every tree decoding step when needed. But if I'm not mistaken, in code, ctxs are passed before tree decoding. I wonder where the code realize this retrieval?