Searched defs:logits_layout (Results 1 – 1 of 1) sorted by relevance
102 const Layout& logits_layout, in ComputeExpAndSum()168 const Layout& logits_layout, in ComputeShardedSoftmax()