Searched defs:TopKTypeConfig (Results 1 – 1 of 1) sorted by relevance
11 struct TopKTypeConfig {}; struct14 struct TopKTypeConfig<float> { struct15 typedef uint32_t RadixType;27 static inline __device__ RadixType convert(float v) { in convert()34 static inline __device__ float deconvert(RadixType v) { in deconvert()42 struct TopKTypeConfig<uint8_t> { struct43 typedef uint32_t RadixType;45 static inline __device__ RadixType convert(uint8_t v) { in convert()49 static inline __device__ uint8_t deconvert(RadixType v) { in deconvert()55 struct TopKTypeConfig<int8_t> { struct[all …]