Searched defs:WARP_BATCH (Results 1 – 1 of 1) sorted by relevance
74 constexpr int WARP_BATCH = (next_power_of_two <= 128) ? 2 : 1; in softmax_warp_forward() local221 constexpr int WARP_BATCH = (next_power_of_two <= 128) ? 2 : 1; in softmax_warp_backward() local