Searched refs:GetMaxWorkGroupSizeForZ (Results 1 – 4 of 4) sorted by relevance
56 z <= gpu_info.GetMaxWorkGroupSizeForZ()) { in GenerateWorkGroupSizesXYMultipleOf()82 z <= gpu_info.GetMaxWorkGroupSizeForZ() && in GenerateWorkGroupSizesXMultipleOf()98 max_wg_size.z = gpu_info.GetMaxWorkGroupSizeForZ(); in GetWorkGroupsAlignedToGrid()306 max_z_size = std::min(max_z_size, gpu_info.GetMaxWorkGroupSizeForZ()); in GetPossibleWorkGroupsConv()
40 while (wg->z > gpu_info.GetMaxWorkGroupSizeForZ()) { in MaybeShrinkWorkgroup()
352 int GetMaxWorkGroupSizeForZ() const;
556 int GpuInfo::GetMaxWorkGroupSizeForZ() const { in GetMaxWorkGroupSizeForZ() function in tflite::gpu::GpuInfo