Class NthElementGPU

Class Documentation

class NthElementGPU

Public Functions

NthElementGPU()
NthElementGPU(const NthElementGPU &copy)
NthElementGPU(size_t maxBeamSize, size_t maxBatchSize, DeviceId deviceId)
~NthElementGPU()
void getNBestList(Tensor scores, size_t N, std::vector<float> &outCosts, std::vector<unsigned> &outKeys, const bool isFirst)