Searched defs:batch_element_count (Results 1 – 1 of 1) sorted by relevance
104 int batch_element_count = (i >= local_batches) ? 0 : element_count; in softmax_warp_forward() local119 int batch_element_count = (i >= local_batches) ? 0 : element_count; in softmax_warp_forward() local150 int batch_element_count = (i >= local_batches) ? 0 : element_count; in softmax_warp_forward() local252 int batch_element_count = (i >= local_batches) ? 0 : element_count; in softmax_warp_backward() local