Function marian::gpu::TopK

• Defined in File topk.cu

Function Documentation

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