Searched defs:logits_buffer (Results 1 – 1 of 1) sorted by relevance
70 static uint64_t argmax(const void* logits_buffer, const size_t vocab_size) { in argmax()85 const void* logits_buffer, in argmax()