Searched defs:next_power_of_two (Results 1 – 5 of 5) sorted by relevance
71 constexpr int next_power_of_two = 1 << log2_elements; in softmax_warp_forward() local218 constexpr int next_power_of_two = 1 << log2_elements; in softmax_warp_backward() local310 const int next_power_of_two = 1 << log2_elements; in dispatch_softmax_forward() local360 const int next_power_of_two = 1 << log2_elements; in dispatch_softmax_backward() local
190 int32_t next_power_of_two = absl::bit_ceil<uint64_t>(width); in CreateTopKUnique() local268 uint32_t next_power_of_two = absl::bit_ceil<uint64_t>(width); in CreateMakeUnique() local
23 static int next_power_of_two(int N) { in next_power_of_two() function
149 static unsigned next_power_of_two(unsigned x) in next_power_of_two() function
322 uint32_t next_power_of_two(uint32_t v) in next_power_of_two() function