Searched refs:GetMaxWorkGroupTotalSize (Results 1 – 4 of 4) sorted by relevance
51 while (CalculateProduct(*wg) > gpu_info.GetMaxWorkGroupTotalSize()) { in MaybeShrinkWorkgroup()
104 std::min(static_cast<uint32_t>(ctx.gpu_info->GetMaxWorkGroupTotalSize()), in GenerateSubgroupBasedMean()
353 int GetMaxWorkGroupTotalSize() const;
572 int GpuInfo::GetMaxWorkGroupTotalSize() const { in GetMaxWorkGroupTotalSize() function in tflite::gpu::GpuInfo