Searched refs:task_size_per_cu (Results 1 – 2 of 2) sorted by relevance
85 const float task_size_per_cu = in GetRecommendedBlockSizeForConv() local144 if (task_size_per_cu <= threshold_1) { in GetRecommendedBlockSizeForConv()146 } else if (task_size_per_cu <= threshold_2) { in GetRecommendedBlockSizeForConv()148 } else if (task_size_per_cu <= threshold_4) { in GetRecommendedBlockSizeForConv()
1094 float task_size_per_cu = in GuessBestParams() local1098 float threads_per_cu = task_size_per_cu / block_size; in GuessBestParams()