Searched refs:GetComputeUnitsCount (Results 1 – 6 of 6) sorted by relevance
161 int GetComputeUnitsCount() const;346 int GetComputeUnitsCount() const;
273 int AppleInfo::GetComputeUnitsCount() const { in GetComputeUnitsCount() function in tflite::gpu::AppleInfo514 int GpuInfo::GetComputeUnitsCount() const { in GetComputeUnitsCount() function in tflite::gpu::GpuInfo519 return apple_info.GetComputeUnitsCount(); in GetComputeUnitsCount()
86 task_size / static_cast<float>(gpu_info.GetComputeUnitsCount()); in GetRecommendedBlockSizeForConv()
140 gpu_info.GetComputeUnitsCount() <= 4) { in GetBestParams()
679 const int cu_count = apple_info.GetComputeUnitsCount(); in GetRecommendedBlockSize()
1095 static_cast<float>(task_size) / gpu_info.GetComputeUnitsCount(); in GuessBestParams()