Class NthElementGPU

• Defined in File nth_element.cu

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)