LeelaChessZero / lc0

The rewritten engine, originally for tensorflow. Now all other backends have been ported here.
GNU General Public License v3.0
2.36k stars 519 forks source link

Allow overriding the set of output nodes in ONNX2XLA #1996

Closed mooskagh closed 3 months ago

mooskagh commented 3 months ago

…d XLA, and only compute the heads that are needed in the XLA backend.

This is extracted from https://github.com/LeelaChessZero/lc0/pull/1986.