Module embedding_lookup 

embedding_lookup op-diff harness — see crate::op_diff.

Gather n rows from a [vocab, dim] table by token id. Exact (no arithmetic), so accelerator NMSE should be ~0.

Structs

EmbeddingLookupOp