Searched refs:GetMaxWorkGroupSizeForY (Results 1 – 5 of 5) sorted by relevance
55 y <= gpu_info.GetMaxWorkGroupSizeForY() && in GenerateWorkGroupSizesXYMultipleOf()81 y <= gpu_info.GetMaxWorkGroupSizeForY() && in GenerateWorkGroupSizesXMultipleOf()97 max_wg_size.y = gpu_info.GetMaxWorkGroupSizeForY(); in GetWorkGroupsAlignedToGrid()
36 while (wg->y > gpu_info.GetMaxWorkGroupSizeForY()) { in MaybeShrinkWorkgroup()
100 const uint32_t max_wg_size_y = ctx.gpu_info->GetMaxWorkGroupSizeForY(); in GenerateSubgroupBasedMean()
351 int GetMaxWorkGroupSizeForY() const;
540 int GpuInfo::GetMaxWorkGroupSizeForY() const { in GetMaxWorkGroupSizeForY() function in tflite::gpu::GpuInfo