Searched defs:WARP_SIZE (Results 1 – 1 of 1) sorted by relevance
72 … constexpr int WARP_SIZE = (next_power_of_two < C10_WARP_SIZE) ? next_power_of_two : C10_WARP_SIZE; in softmax_warp_forward() local219 … constexpr int WARP_SIZE = (next_power_of_two < C10_WARP_SIZE) ? next_power_of_two : C10_WARP_SIZE; in softmax_warp_backward() local