Searched refs:GetComputeUnitsCount (Results 1 – 6 of 6) sorted by relevance
102 int GetComputeUnitsCount() const { in GetComputeUnitsCount() function171 int GetComputeUnitsCount() const;376 int GetComputeUnitsCount() const;
277 int AppleInfo::GetComputeUnitsCount() const { in GetComputeUnitsCount() function in tflite::gpu::AppleInfo548 int GpuInfo::GetComputeUnitsCount() const { in GetComputeUnitsCount() function in tflite::gpu::GpuInfo553 return apple_info.GetComputeUnitsCount(); in GetComputeUnitsCount()556 return amd_info.GetComputeUnitsCount(); in GetComputeUnitsCount()
86 task_size / static_cast<float>(gpu_info.GetComputeUnitsCount()); in GetRecommendedBlockSizeForConv()
181 gpu_info.GetComputeUnitsCount() <= 4) { in GetBestParams()
679 const int cu_count = apple_info.GetComputeUnitsCount(); in GetRecommendedBlockSize()
1119 static_cast<float>(task_size) / gpu_info.GetComputeUnitsCount(); in GuessBestParams()