Function marian::cpu::TopK

• Defined in File topk.cpp

Function Documentation

void marian::cpu::TopK(Tensor outVal, Tensor outInd, Ptr<Allocator>, const Tensor in, int k, int axis, bool descending)