Searched refs:GetComputeUnitsCount (Results 1 – 6 of 6) sorted by relevance
105 int GetComputeUnitsCount() const { in GetComputeUnitsCount() function144 int GetComputeUnitsCount() const;199 int GetComputeUnitsCount() const;457 int GetComputeUnitsCount() const;
263 int AdrenoInfo::GetComputeUnitsCount() const { in GetComputeUnitsCount() function in tflite::gpu::AdrenoInfo408 int AppleInfo::GetComputeUnitsCount() const { in GetComputeUnitsCount() function in tflite::gpu::AppleInfo838 int GpuInfo::GetComputeUnitsCount() const { in GetComputeUnitsCount() function in tflite::gpu::GpuInfo843 return apple_info.GetComputeUnitsCount(); in GetComputeUnitsCount()846 if (amd_info.GetComputeUnitsCount() != 0) { in GetComputeUnitsCount()847 return amd_info.GetComputeUnitsCount(); in GetComputeUnitsCount()854 return adreno_info.GetComputeUnitsCount(); in GetComputeUnitsCount()
140 task_size / static_cast<float>(gpu_info.GetComputeUnitsCount()); in GetRecommendedBlockSizeForConv()
352 alu_fp32_gflops_per_cu * device.GetInfo().apple_info.GetComputeUnitsCount();355 alu_fp16_gflops_per_cu * device.GetInfo().apple_info.GetComputeUnitsCount();
1195 const int cu_count = apple_info.GetComputeUnitsCount(); in GetRecommendedBlockSize()1506 static_cast<float>(task_size) / gpu_info.GetComputeUnitsCount(); in GuessBestParams()1606 static_cast<float>(task_size) / gpu_info.GetComputeUnitsCount(); in GuessBestParams()1691 task_size / gpu_info.GetComputeUnitsCount() / wave_size; in GuessBestParams()
554 const double task_size_per_cu = task_size / gpu_info.GetComputeUnitsCount(); in IsGoodTaskSizeForAppleConvSimd()